Commit b4fd2a77 authored by miaecle's avatar miaecle
Browse files

update unit tests

parent 5c5a918c

deepchem/models/tests/test_overfit.py

+2 −0
Original line number Diff line number Diff line
@@ -725,6 +725,8 @@ class TestOverfit(test_util.TensorFlowTestCase): 


    n_feat = 75

    batch_size = 10

    transformer = dc.trans.DAGTransformer(max_atoms=50)

    dataset = transformer.transform(dataset)



    graph = dc.nn.SequentialDAGGraph(

        n_feat, batch_size=batch_size, max_atoms=50)

CRA Git | Maintained and supported by SUSTech CRA and CCSE