Commit 0088297c authored by VIGNESHinZONE's avatar VIGNESHinZONE
Browse files

fixes

parent 0d16dcc1
Loading
Loading
Loading
Loading
+1 −1
Original line number Diff line number Diff line
@@ -5,11 +5,11 @@ Gathers all models in one place for convenient imports
from deepchem.models.models import Model
from deepchem.models.multitask import SingletaskToMultitask
from deepchem.models.wandblogger import WandbLogger
from deepchem.models.callbacks import ValidationCallback

# Tensorflow Depedency Models
try:
  from deepchem.models.keras_model import KerasModel
  from deepchem.models.callbacks import ValidationCallback

  from deepchem.models.IRV import MultitaskIRVClassifier
  from deepchem.models.robust_multitask import RobustMultitaskClassifier
+2 −2
Original line number Diff line number Diff line
@@ -6,13 +6,13 @@ import sklearn
from deepchem.utils.evaluate import Evaluator
from deepchem.utils.evaluate import GeneratorEvaluator
try:
  import tensorflow as tf
  import tensorflow as tf  # noqa
  has_tensorflow = True
except:
  has_tensorflow = False

try:
  import torch
  import torch  # noqa
  has_pytorch = True
except:
  has_pytorch = False