Merge pull request #2567 from VIGNESHinZONE/tf-free (d62afe44) · Commits · 钟慕尧 / deepchem

.github/workflows/jax_setup.yml

+0 −1

Original line number	Diff line number	Diff line
		@@ -29,7 +29,6 @@ jobs:
		- name: Build DeepChem
		run: \|
		python -m pip install --upgrade pip
		pip install tensorflow==2.4
		pip install -e '.[jax]'
		- name: Import checking
		run: python -c "import deepchem; import jax;"

deepchem/metalearning/init.py

+4 −2

Original line number	Diff line number	Diff line
		# flake8: noqa

		try:
		from deepchem.metalearning.maml import MAML, MetaLearner
		except ModuleNotFoundError:
		pass

deepchem/models/init.py

+28 −21

Original line number	Diff line number	Diff line
		@@ -3,10 +3,13 @@ Gathers all models in one place for convenient imports
		"""
		# flake8: noqa
		from deepchem.models.models import Model
		from deepchem.models.wandblogger import WandbLogger

		# Tensorflow Depedency Models
		try:
		from deepchem.models.keras_model import KerasModel
		from deepchem.models.multitask import SingletaskToMultitask
		from deepchem.models.callbacks import ValidationCallback
		from deepchem.models.wandblogger import WandbLogger

		from deepchem.models.IRV import MultitaskIRVClassifier
		from deepchem.models.robust_multitask import RobustMultitaskClassifier
		@@ -22,6 +25,8 @@ from deepchem.models.cnn import CNN
		from deepchem.models.text_cnn import TextCNNModel
		from deepchem.models.atomic_conv import AtomicConvModel
		from deepchem.models.chemnet_models import Smiles2Vec, ChemCeption
		except ModuleNotFoundError:
		pass

		# scikit-learn model
		from deepchem.models.sklearn_models import SklearnModel
		@@ -49,7 +54,9 @@ from deepchem.models.gbdt_models.gbdt_model import XGBoostModel
		########################################################################################
		# Compatibility imports for renamed TensorGraph models. Remove below with DeepChem 3.0.
		########################################################################################

		try:
		from deepchem.models.text_cnn import TextCNNTensorGraph
		from deepchem.models.graph_models import WeaveTensorGraph, DTNNTensorGraph, DAGTensorGraph, GraphConvTensorGraph, MPNNTensorGraph
		from deepchem.models.IRV import TensorflowMultitaskIRVClassifier
		except ModuleNotFoundError:
		pass

deepchem/rl/init.py

+5 −3

Original line number	Diff line number	Diff line
		"""Interface for reinforcement learning."""

		try:
		from deepchem.rl.a2c import A2C # noqa: F401
		from deepchem.rl.ppo import PPO # noqa: F401
		except ModuleNotFoundError:
		pass


		class Environment(object):

deepchem/trans/init.py

+0 −1

Original line number	Diff line number	Diff line
		@@ -15,7 +15,6 @@ from deepchem.trans.transformers import PowerTransformer
		from deepchem.trans.transformers import CoulombFitTransformer
		from deepchem.trans.transformers import IRVTransformer
		from deepchem.trans.transformers import DAGTransformer
		from deepchem.trans.transformers import ANITransformer
		from deepchem.trans.transformers import MinMaxTransformer
		from deepchem.trans.transformers import FeaturizationTransformer
		from deepchem.trans.transformers import ImageTransformer

Admin message