Merge pull request graphnet-team#604 from RasmusOrsoe/warning_once_fix

Warning once fix
RasmusOrsoe · Sep 25, 2023 · 9fb02f2 · 9fb02f2
2 parents 822940c + 68fe515
commit 9fb02f2
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 6 deletions.
diff --git a/src/graphnet/models/graphs/edges/edges.py b/src/graphnet/models/graphs/edges/edges.py
@@ -26,7 +26,7 @@ def forward(self, graph: Data) -> Data:
             graph: a graph with edges
         """
         if graph.edge_index is not None:
-            self.warnonce(
+            self.warning_once(
                 "GraphBuilder received graph with pre-existing "
                 "structure. Will overwrite."
             )

diff --git a/src/graphnet/models/graphs/graph_definition.py b/src/graphnet/models/graphs/graph_definition.py
@@ -148,7 +148,8 @@ def forward(  # type: ignore
         if self._edge_definition is not None:
             graph = self._edge_definition(graph)
         else:
-            self.warnonce(
+
+            self.warning_once(
                 """No EdgeDefinition provided. 
                 Graphs will not have edges defined!"""  # noqa
             )
@@ -305,10 +306,11 @@ def _add_features_individually(
             if feature not in ["x"]:  # reserved for node features.
                 graph[feature] = graph.x[:, index].detach()
             else:
-                self.warnonce(
-                    """Cannot assign graph['x']. This field is reserved 
-                    for node features. Please rename your input feature."""  # noqa
-                )
+                self.warning_once(
+                    """Cannot assign graph['x']. This field is reserved for
+                      node features. Please rename your input feature."""
+                )  # noqa
+
         return graph
 
     def _add_custom_labels(