Fixed import error when pytorch is not installed (186c1689) · Commits · 钟慕尧 / deepchem

deepchem/models/init.py

+1 −3

Original line number	Diff line number	Diff line
		@@ -7,9 +7,6 @@ from deepchem.models.keras_model import KerasModel
		from deepchem.models.multitask import SingletaskToMultitask
		from deepchem.models.callbacks import ValidationCallback

		from deepchem.models.fcnet import MultitaskRegressor
		from deepchem.models.fcnet import MultitaskClassifier
		from deepchem.models.fcnet import MultitaskFitTransformRegressor
		from deepchem.models.IRV import MultitaskIRVClassifier
		from deepchem.models.robust_multitask import RobustMultitaskClassifier
		from deepchem.models.robust_multitask import RobustMultitaskRegressor
		@@ -38,6 +35,7 @@ try:
		from deepchem.models.torch_models import GCN, GCNModel
		from deepchem.models.torch_models import LCNN, LCNNModel
		from deepchem.models.torch_models import Pagtn, PagtnModel
		from deepchem.models.fcnet import MultitaskRegressor, MultitaskClassifier, MultitaskFitTransformRegressor
		except ModuleNotFoundError:
		pass