test: update delta checkpoint and chained delta tests

- Modify integration test for delta checkpoint flow
- Update unit tests for chained deltas and checkpoint cleanup

Author: erfanMhi

tests/integration/infrastructure/test_delta_checkpoint_flow.py

@@ -297,7 +297,7 @@ def test_metadata_is_delta(self) -> None:
             window=110,
             file_manifest={},
             checkpoint_type="DELTA",
-            base_window=100,
+            anchor_window=100,
             weights_hash="abc123",
         )
         assert delta_metadata.is_delta()
@@ -308,7 +308,7 @@ def test_metadata_serialization(self) -> None:
             window=110,
             file_manifest={"delta_sparse.safetensors": "hash123"},
             checkpoint_type="DELTA",
-            base_window=100,
+            anchor_window=100,
             weights_hash="abcdef1234567890",
         )
 
@@ -319,7 +319,7 @@ def test_metadata_serialization(self) -> None:
         # Deserialize
         loaded = json.loads(json_str)
         assert loaded["checkpoint_type"] == "DELTA"
-        assert loaded["base_window"] == 100
+        assert loaded["anchor_window"] == 100
         assert loaded["weights_hash"] == "abcdef1234567890"
 
 

tests/unit/infrastructure/test_chained_deltas.py

@@ -149,13 +149,13 @@ class TestRetentionPolicyChainedDeltas:
 
     def test_keeps_entire_active_chain(self) -> None:
         """Should keep all windows from current anchor to now."""
-        from grail.trainer.checkpoint_publisher import _compute_keep_windows
+        from grail.shared.retention_utils import compute_retention_windows
 
-        with patch("grail.trainer.checkpoint_publisher.DELTA_BASE_INTERVAL", 5):
-            with patch("grail.trainer.checkpoint_publisher.WINDOW_LENGTH", 30):
+        with patch("grail.shared.retention_utils.DELTA_BASE_INTERVAL", 5):
+            with patch("grail.shared.retention_utils.WINDOW_LENGTH", 30):
                 # Anchor stride = 5 * 30 = 150
                 # Current window 180: anchor at 150
-                result = _compute_keep_windows(180)
+                result = compute_retention_windows(180)
 
                 # Should keep: 150, 180 (chain from 150 to 180)
                 # Also 0 (previous anchor) and 0-150 chain
@@ -164,14 +164,14 @@ def test_keeps_entire_active_chain(self) -> None:
 
     def test_keeps_previous_anchor_chain(self) -> None:
         """Should keep previous anchor's chain for transition."""
-        from grail.trainer.checkpoint_publisher import _compute_keep_windows
+        from grail.shared.retention_utils import compute_retention_windows
 
-        with patch("grail.trainer.checkpoint_publisher.DELTA_BASE_INTERVAL", 3):
-            with patch("grail.trainer.checkpoint_publisher.WINDOW_LENGTH", 30):
+        with patch("grail.shared.retention_utils.DELTA_BASE_INTERVAL", 3):
+            with patch("grail.shared.retention_utils.WINDOW_LENGTH", 30):
                 # Anchor stride = 3 * 30 = 90
                 # Current window 120: anchor at 90
                 # Previous anchor at 0
-                result = _compute_keep_windows(120)
+                result = compute_retention_windows(120)
 
                 # Current chain: 90, 120
                 assert 90 in result
@@ -190,8 +190,8 @@ def test_consumer_retention_keeps_chain(self) -> None:
             keep_limit=3,
         )
 
-        with patch("grail.infrastructure.checkpoint_consumer.DELTA_BASE_INTERVAL", 5):
-            with patch("grail.infrastructure.checkpoint_consumer.WINDOW_LENGTH", 30):
+        with patch("grail.shared.retention_utils.DELTA_BASE_INTERVAL", 5):
+            with patch("grail.shared.retention_utils.WINDOW_LENGTH", 30):
                 result = manager._compute_keep_windows(180)
 
                 # Should keep chain from anchor (150) to now (180)