fix(graphs): support torch v2.6 (#122)

ecmwf · Feb 10, 2025 · dfef377 · dfef377
1 parent beb8c69
commit dfef377
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/graphs/src/anemoi/graphs/describe.py b/graphs/src/anemoi/graphs/describe.py
@@ -24,7 +24,7 @@ class GraphDescriptor:
 
     def __init__(self, path: Union[str, Path], **kwargs):
         self.path = path
-        self.graph = torch.load(self.path)
+        self.graph = torch.load(self.path, weights_only=False)
 
     @property
     def total_size(self):

diff --git a/graphs/src/anemoi/graphs/inspect.py b/graphs/src/anemoi/graphs/inspect.py
@@ -49,7 +49,7 @@ def __init__(
         **kwargs,
     ):
         self.path = path
-        self.graph = torch.load(self.path)
+        self.graph = torch.load(self.path, weights_only=False)
         self.output_path = output_path
         self.show_attribute_distributions = show_attribute_distributions
         self.show_nodes = show_nodes