From 4019d5526562c0c53aa56da587f43eec9e1fd1df Mon Sep 17 00:00:00 2001 From: Bharath Ramsundar Date: Wed, 21 Oct 2020 18:37:53 -0700 Subject: [PATCH] Fixing scscore --- deepchem/models/tests/test_scscore.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/deepchem/models/tests/test_scscore.py b/deepchem/models/tests/test_scscore.py index 89be00886c..1576ae86d4 100644 --- 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)