Commit 591ae6e8 authored by Michelle Gill's avatar Michelle Gill
Browse files

DTNNTensorGraph deprecation setup

parent b6a0cff5

deepchem/models/__init__.py

+1 −1
Original line number Diff line number Diff line
@@ -28,4 +28,4 @@ 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 WeaveTensorGraph, GraphConvTensorGraph
 
 No newline at end of file
 
from deepchem.models.tensorgraph.models.graph_models import WeaveTensorGraph, DTNNTensorGraph, GraphConvTensorGraph
 
 No newline at end of file

deepchem/models/tensorgraph/models/graph_models.py

+11 −0
Original line number Diff line number Diff line
@@ -1172,3 +1172,14 @@ class WeaveTensorGraph(WeaveModel): 
  def __init__(self, *args, **kwargs):



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





class DTNNTensorGraph(DTNNModel):



  warnings.warn(

      TENSORGRAPH_DEPRECATION.format("DTNNTensorGraph", "DTNNModel"),

      FutureWarning)



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



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

CRA Git | Maintained and supported by SUSTech CRA and CCSE