Commit d37afe85 authored by Bharath Ramsundar's avatar Bharath Ramsundar
Browse files

Removing some imports

parent 712e51c1
Loading
Loading
Loading
Loading
+0 −4
Original line number Diff line number Diff line
@@ -15,10 +15,6 @@ import deepchem as dc
from deepchem.nn import model_ops
from deepchem.utils.save import log
from deepchem.metrics import to_one_hot, from_one_hot
from deepchem.models.tensorflow_models import TensorflowGraph
from deepchem.models.tensorflow_models import TensorflowGraphModel
from deepchem.models.tensorflow_models import TensorflowClassifier
from deepchem.models.tensorflow_models import TensorflowRegressor
from deepchem.metrics import to_one_hot

from deepchem.models.tensorgraph.tensor_graph import TensorGraph, TFWrapper
+3 −4
Original line number Diff line number Diff line
@@ -53,10 +53,9 @@ from deepchem.models.tensorgraph.layers import Variable
from deepchem.models.tensorgraph.layers import VinaFreeEnergy
from deepchem.models.tensorgraph.layers import WeightedError
from deepchem.models.tensorgraph.layers import WeightedLinearCombo

from deepchem.models.tensorflow_models.IRV import IRVLayer
from deepchem.models.tensorflow_models.IRV import IRVRegularize
from deepchem.models.tensorflow_models.IRV import Slice
from deepchem.models.tensorgraph.IRV import IRVLayer
from deepchem.models.tensorgraph.IRV import IRVRegularize
from deepchem.models.tensorgraph.IRV import Slice


class TestLayers(test_util.TensorFlowTestCase):
+1 −1
Original line number Diff line number Diff line
@@ -12,7 +12,7 @@ from deepchem.models.tensorgraph.layers import Feature, Conv1D, Dense, Flatten,
  Conv3D, MaxPool3D, Conv2DTranspose, Conv3DTranspose, \
  LSTMStep, AttnLSTMEmbedding, IterRefLSTMEmbedding, GraphEmbedPoolLayer, GraphCNN, Cast
from deepchem.models.tensorgraph.symmetry_functions import AtomicDifferentiatedDense
from deepchem.models.tensorflow_models.IRV import IRVLayer, IRVRegularize, Slice
from deepchem.models.tensorgraph.IRV import IRVLayer, IRVRegularize, Slice


def test_Conv1D_pickle():