Fixing scscore

deepchem · Oct 22, 2020 · 4019d55 · 4019d55
1 parent 5768421
commit 4019d55
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/deepchem/models/tests/test_scscore.py b/deepchem/models/tests/test_scscore.py
@@ -1,6 +1,6 @@
 import unittest
-
-import deepchem
+import tempfile
+import deepchem as dc
 import numpy as np
 
 
@@ -16,9 +16,9 @@ def test_overfit_scscore(self):
 
     X = np.random.rand(n_samples, 2, n_features)
     y = np.random.randint(2, size=(n_samples, n_tasks))
-    dataset = deepchem.data.NumpyDataset(X, y)
+    dataset = dc.data.NumpyDataset(X, y)
 
-    model = deepchem.models.ScScoreModel(n_features, dropouts=0)
+    model = dc.models.ScScoreModel(n_features, dropouts=0)
 
     model.fit(dataset, nb_epoch=100)
     pred = model.predict(dataset)