Commit 48aa329a authored by Michelle Gill's avatar Michelle Gill
Browse files

Deprecation subclass with warning and imports

parent ce5f7574

deepchem/models/__init__.py

+4 −0
Original line number Diff line number Diff line
@@ -25,3 +25,7 @@ from deepchem.models.tensorgraph.models.gan import GAN, WGAN 
from deepchem.models.tensorgraph.models.text_cnn import TextCNNTensorGraph

from deepchem.models.tensorgraph.sequential import Sequential

from deepchem.models.tensorgraph.models.sequence_dnn import SequenceDNN



#################### Compatibility imports for renamed TensorGraph models. Remove below with DeepChem 3.0. ####################



from deepchem.models.tensorgraph.models.graph_models import GraphConvTensorGraph
 
 No newline at end of file

deepchem/models/tensorgraph/models/graph_models.py

+15 −0
Original line number Diff line number Diff line
@@ -1143,3 +1143,18 @@ class MPNNTensorGraph(TensorGraph): 


  def predict_on_generator(self, generator, transformers=[]):

    return self.predict_proba_on_generator(generator, transformers)





#################### Deprecation warnings for renamed TensorGraph models ####################



import warnings



class GraphConvTensorGraph(GraphConvModel):



  def __init__(self, *args, **kwargs):



    warnings.warn("GraphConvTensorGraph is deprecated and has been renamed to GraphConvModel. "

                    "Will be removed in DeepChem 3.0.", DeprecationWarning)



    super(GraphConvTensorGraph, self).__init__(*args, **kwargs)

CRA Git | Maintained and supported by SUSTech CRA and CCSE