update tests with new functionality

henrique · henrique · commit 64ae116bf3f6 · 2026-03-16T13:06:45.000Z
diff --git a/tests/test_ckpt.py b/tests/test_ckpt.py
@@ -243,7 +243,7 @@ def test_load_missing_epoch(self, tmp_path: Path):
         )
 
         metadata: Optional[Dict[str, Any]] = load_run_metadata(tmp_path, epoch=99)
-        assert metadata == {}
+        assert metadata == None
 
     def test_metadata_accumulates(self, tmp_path: Path):
         """Tests that multiple saves accumulate in the same file."""

Original file line number	Diff line number	Diff line change
`@@ -243,7 +243,7 @@ def test_load_missing_epoch(self, tmp_path: Path):`
`243`	`243`	`)`
`244`	`244`
`245`	`245`	`metadata: Optional[Dict[str, Any]] = load_run_metadata(tmp_path, epoch=99)`
`246`		`- assert metadata == {}`
	`246`	`+ assert metadata == None`
`247`	`247`
`248`	`248`	`def test_metadata_accumulates(self, tmp_path: Path):`
`249`	`249`	`"""Tests that multiple saves accumulate in the same file."""`