Commit 3cb15939 authored by Michelle Gill's avatar Michelle Gill
Browse files

Move deprecation to constructor

parent 47cfb41a
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):

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

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

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

    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)