Commit 6e7e8e46 authored by Michelle Gill's avatar Michelle Gill
Browse files

MPNNTensorGraph Deprecation

parent ada0d807
Loading
Loading
Loading
Loading
+11 −0
Original line number Diff line number Diff line
@@ -1194,3 +1194,14 @@ class PetroskiSuchTensorGraph(PetroskiSuchModel):
  def __init__(self, *args, **kwargs):

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


class MPNNTensorGraph(MPNNModel):

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

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

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