fix(graphs): make exception agnostic of zarr version (#152)

* fix(graphs): make expection agnostic of zarr version
ecmwf · Feb 24, 2025 · f26adc9 · f26adc9
1 parent 505d084
commit f26adc9
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/graphs/tests/nodes/test_zarr.py b/graphs/tests/nodes/test_zarr.py
@@ -9,7 +9,6 @@
 
 import pytest
 import torch
-import zarr
 from torch_geometric.data import HeteroData
 
 from anemoi.graphs.nodes.attributes import SphericalAreaWeights
@@ -29,7 +28,7 @@ def test_init(mocker, mock_zarr_dataset):
 def test_fail():
     """Test ZarrDatasetNodes with invalid dataset."""
     node_builder = from_file.ZarrDatasetNodes("invalid_path.zarr", name="test_nodes")
-    with pytest.raises(zarr.errors.PathNotFoundError):
+    with pytest.raises(Exception):
         node_builder.update_graph(HeteroData())