Skip to content

Commit 73e87c5

Browse files
committed
fix for torch v2.6
1 parent c4f2c70 commit 73e87c5

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

Diff for: graphs/src/anemoi/graphs/describe.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,7 @@ class GraphDescriptor:
2424

2525
def __init__(self, path: Union[str, Path], **kwargs):
2626
self.path = path
27-
self.graph = torch.load(self.path)
27+
self.graph = torch.load(self.path, weights_only=False)
2828

2929
@property
3030
def total_size(self):

Diff for: graphs/src/anemoi/graphs/inspect.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -49,7 +49,7 @@ def __init__(
4949
**kwargs,
5050
):
5151
self.path = path
52-
self.graph = torch.load(self.path)
52+
self.graph = torch.load(self.path, weights_only=False)
5353
self.output_path = output_path
5454
self.show_attribute_distributions = show_attribute_distributions
5555
self.show_nodes = show_nodes

0 commit comments

Comments
 (0)