Commit f30efbbf authored by miaecle's avatar miaecle
Browse files

fix test failure

parent ec395116

deepchem/models/tensorgraph/models/test_graph_models.py

+5 −1
Original line number Diff line number Diff line
@@ -19,7 +19,11 @@ class TestGraphModels(unittest.TestCase): 
                  featurizer='GraphConv',

                  num_tasks=2):

    data_points = 10

    if mode == 'classification':

      tasks, all_dataset, transformers = load_bace(featurizer)

    else:

      tasks, all_dataset, transformers = load_delaney(featurizer)



    train, valid, test = all_dataset

    for i in range(1, num_tasks):

      tasks.append("random_task")

CRA Git | Maintained and supported by SUSTech CRA and CCSE