Update checkpoint name unit tests

Author: havardhhaugen

bris/model/interpolator.py

@@ -692,7 +692,6 @@ def forward_interp(self, x: torch.Tensor, **kwargs) -> torch.Tensor:
             )[self.ds_name_interp]
         except AttributeError:
             # Backward compatibility with models that do not use kwargs:
-            x = list(x.values())[0]
             try:
                 return self.interpolator(
                     x, model_comm_group=self.model_comm_group, **kwargs

tests/test_checkpoints.py

@@ -6,7 +6,7 @@
 
 
 def test_metadata():
-    filename = os.path.dirname(os.path.abspath(__file__)) + "/files/checkpoint.ckpt"
+    filename = os.path.dirname(os.path.abspath(__file__)) + "/files/checkpoint_single.ckpt"
     checkpoint = bris.checkpoint.Checkpoint(path=filename)
 
     n2i = {

tests/test_inspect.py

@@ -115,7 +115,7 @@ def test_get_required_variables():
         "skt",
         "w_850",
     ]
-    checkpoint_simple = bris.checkpoint.Checkpoint("tests/files/checkpoint.ckpt")
+    checkpoint_simple = bris.checkpoint.Checkpoint("tests/files/checkpoint_single.ckpt")
     required_simple = bris.inspect.get_required_variables(checkpoint_simple)["data"]
     for var in expected_simple:
         assert var in required_simple, (
@@ -143,12 +143,12 @@ def test_get_required_variables():
 
 def test_check_module_versions():
     """This depends on the current venv, so just test it doesn't crash"""
-    checkpoint = bris.checkpoint.Checkpoint("tests/files/checkpoint.ckpt")
+    checkpoint = bris.checkpoint.Checkpoint("tests/files/checkpoint_single.ckpt")
     _bad = bris.inspect.check_module_versions(checkpoint)
 
     # assert "fsspec==2025.2.0" in bad
 
 
 def manual_test_inspect():
     """This depends on the current venv, so just test it doesn't crash"""
-    _status = bris.inspect.inspect(checkpoint_path="tests/files/checkpoint.ckpt")
+    _status = bris.inspect.inspect(checkpoint_path="tests/files/checkpoint_single.ckpt")

tests/test_model.py

@@ -35,7 +35,7 @@ def test_bris_predictor():
         )
 
     checkpoint_path = (
-        os.path.dirname(os.path.abspath(__file__)) + "/files/checkpoint.ckpt"
+        os.path.dirname(os.path.abspath(__file__)) + "/files/checkpoint_single.ckpt"
     )
 
     # Create test config

tests/test_routes.py

@@ -144,6 +144,6 @@ def test_add_checkpoint_name_to_attrs():
             "variables": ["2t", "2d"],
         }
     }
-    test_ckpts = {"testchk": Checkpoint("./tests/files/checkpoint.ckpt")}
+    test_ckpts = {"testchk": Checkpoint("./tests/files/checkpoint_single.ckpt")}
     new_oc = bris.routes.add_checkpoint_name_to_attrs(test_oc, test_ckpts)
     assert "testchk" in new_oc["netcdf"]["global_attributes"]["source"]