Commit a3ac57d5 authored by Michelle Gill's avatar Michelle Gill
Browse files

DAGTensorGraph Deprecation

parent f20016c8

deepchem/models/__init__.py

+1 −1
Original line number Diff line number Diff line
@@ -29,4 +29,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.text_cnn import TextCNNTensorGraph

from deepchem.models.tensorgraph.models.graph_models import WeaveTensorGraph, DTNNTensorGraph, GraphConvTensorGraph, MPNNTensorGraphModel
 
from deepchem.models.tensorgraph.models.graph_models import WeaveTensorGraph, DTNNTensorGraph, DAGTensorGraph, GraphConvTensorGraph, MPNNTensorGraphModel

deepchem/models/tensorgraph/models/graph_models.py

+11 −0
Original line number Diff line number Diff line
@@ -1185,6 +1185,17 @@ class DTNNTensorGraph(DTNNModel): 
    super(DTNNModel, self).__init__(*args, **kwargs)





class DAGTensorGraph(DAGModel):



  warnings.warn(

      TENSORGRAPH_DEPRECATION.format("DAGTensorGraph", "DAGModel"),

      FutureWarning)



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



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





class PetroskiSuchTensorGraph(PetroskiSuchModel):



  warnings.warn(

CRA Git | Maintained and supported by SUSTech CRA and CCSE