Unverified Commit 2d7990a9 authored by Karl Leswing's avatar Karl Leswing Committed by GitHub
Browse files

Merge pull request #1196 from mlgill/issue1138_mg_remaining_tensor_graph

Move Tensor Graph old model deprecation warning to constructor
parents 47cfb41a 8783e490
Loading
Loading
Loading
Loading
+24 −24
Original line number Diff line number Diff line
@@ -1154,65 +1154,65 @@ TENSORGRAPH_DEPRECATION = "{} is deprecated and has been renamed to {} and will

class GraphConvTensorGraph(GraphConvModel):

  warnings.warn(
      TENSORGRAPH_DEPRECATION.format("GraphConvTensorGraph", "GraphConvModel"),
      FutureWarning)

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

    warnings.warn(
        TENSORGRAPH_DEPRECATION.format("GraphConvTensorGraph",
                                       "GraphConvModel"), FutureWarning)

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


class WeaveTensorGraph(WeaveModel):

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

    warnings.warn(
        TENSORGRAPH_DEPRECATION.format("WeaveTensorGraph", "WeaveModel"),
        FutureWarning)

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

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


class DTNNTensorGraph(DTNNModel):

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

    warnings.warn(
        TENSORGRAPH_DEPRECATION.format("DTNNTensorGraph", "DTNNModel"),
        FutureWarning)

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

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


class DAGTensorGraph(DAGModel):

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

    warnings.warn(
        TENSORGRAPH_DEPRECATION.format("DAGTensorGraph", "DAGModel"),
        FutureWarning)

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

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


class PetroskiSuchTensorGraph(PetroskiSuchModel):

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

    warnings.warn(
        TENSORGRAPH_DEPRECATION.format("PetroskiSuchTensorGraph",
                                       "PetroskiSuchModel"), FutureWarning)

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

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


class MPNNTensorGraph(MPNNModel):

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

    warnings.warn(
        TENSORGRAPH_DEPRECATION.format("MPNNTensorGraph", "MPNNModel"),
        FutureWarning)

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

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