Wrap old class call with deprecation message. (db48a5c8) · Commits · 钟慕尧 / deepchem

deepchem/models/tensorgraph/models/graph_models.py

+6 −6

Original line number	Diff line number	Diff line
		@@ -20,7 +20,7 @@ from deepchem.models.tensorgraph.tensor_graph import TensorGraph
		from deepchem.trans import undo_transforms


		class WeaveTensorGraph(TensorGraph):
		class WeaveTensorModel(TensorGraph):

		def __init__(self,
		n_tasks,
		@@ -52,7 +52,7 @@ class WeaveTensorGraph(TensorGraph):
		self.n_hidden = n_hidden
		self.n_graph_feat = n_graph_feat
		self.mode = mode
		super(WeaveTensorGraph, self).__init__(**kwargs)
		super(WeaveTensorModel, self).__init__(**kwargs)
		self.build_graph()

		def build_graph(self):
		@@ -183,7 +183,7 @@ class WeaveTensorGraph(TensorGraph):
		yield feed_dict

		def predict_on_generator(self, generator, transformers=[], outputs=None):
		out = super(WeaveTensorGraph, self).predict_on_generator(
		out = super(WeaveTensorModel, self).predict_on_generator(
		generator, transformers=[], outputs=outputs)
		if outputs is None:
		outputs = self.outputs
		@@ -1149,12 +1149,12 @@ class MPNNTensorGraph(TensorGraph):

		import warnings

		TENSORGRAPH_DEPRECATION = "{} is deprecated and has been renamed to GraphConvModel and will be removed in DeepChem 3.0."


		class GraphConvTensorGraph(GraphConvModel):

		warnings.warn(
		"GraphConvTensorGraph is deprecated and has been renamed to GraphConvModel and will be removed in DeepChem 3.0.",
		FutureWarning)
		warnings.warn(TENSORGRAPH_DEPRECATION.format("GraphConvTensorGraph"), FutureWarning)

		def __init__(self, args, *kwargs):