@@ -113,8 +113,7 @@ def mock_checkpoint_manager(test_checkpoint_data: dict) -> Mock:
113113 manager .load = AsyncMock (return_value = checkpoint )
114114 manager .save = AsyncMock ()
115115 manager .delete = AsyncMock ()
116- # Some tests may call `load_checkpoint()` instead of `load()`;
117- # make it an alias so both return the same checkpoint object.
116+ # Alias provided for backwards compatibility if needed.
118117 if hasattr (manager , "load_checkpoint" ):
119118 manager .load_checkpoint = manager .load
120119 if hasattr (manager , "validate_checkpoint_compatibility" ):
@@ -336,7 +335,7 @@ async def test_compatible_query_model_change(
336335 new_config .datatype = "float32"
337336
338337 # Get checkpoint metadata
339- checkpoint = await mock_checkpoint_manager .load_checkpoint ()
338+ checkpoint = await mock_checkpoint_manager .load ()
340339
341340 # Create fingerprint from checkpoint metadata
342341 from codeweaver .engine .managers .checkpoint_manager import CheckpointSettingsFingerprint
@@ -382,7 +381,7 @@ async def test_transformable_dimension_reduction(
382381 new_config .dimension = 1024 # Reduced from 2048
383382 new_config .datatype = "float32"
384383
385- checkpoint = await mock_checkpoint_manager .load_checkpoint ()
384+ checkpoint = await mock_checkpoint_manager .load ()
386385
387386 # Create fingerprint from checkpoint metadata
388387 from codeweaver .engine .managers .checkpoint_manager import CheckpointSettingsFingerprint
@@ -430,7 +429,7 @@ async def test_breaking_model_change(
430429 new_config .dimension = 2048
431430 new_config .datatype = "float32"
432431
433- checkpoint = await mock_checkpoint_manager .load_checkpoint ()
432+ checkpoint = await mock_checkpoint_manager .load ()
434433
435434 # Create fingerprint from checkpoint metadata
436435 from codeweaver .engine .managers .checkpoint_manager import CheckpointSettingsFingerprint
@@ -556,7 +555,7 @@ async def test_uses_voyage_3_empirical_data(
556555 new_config .dimension = 512
557556 new_config .datatype = "float32"
558557
559- checkpoint = await mock_checkpoint_manager .load_checkpoint ()
558+ checkpoint = await mock_checkpoint_manager .load ()
560559
561560 # Create fingerprint
562561 from codeweaver .engine .managers .checkpoint_manager import CheckpointSettingsFingerprint
@@ -605,7 +604,7 @@ async def test_falls_back_to_generic_for_unmapped_dimensions(
605604 new_config .dimension = 768 # Uncommon dimension
606605 new_config .datatype = "float32"
607606
608- checkpoint = await mock_checkpoint_manager .load_checkpoint ()
607+ checkpoint = await mock_checkpoint_manager .load ()
609608
610609 # Create fingerprint
611610 from codeweaver .engine .managers .checkpoint_manager import CheckpointSettingsFingerprint
@@ -656,7 +655,7 @@ async def test_handles_very_large_collection(
656655 from codeweaver .engine .services .config_analyzer import ConfigChangeAnalyzer
657656
658657 # Update checkpoint with large vector count
659- checkpoint = await mock_checkpoint_manager .load_checkpoint ()
658+ checkpoint = await mock_checkpoint_manager .load ()
660659 checkpoint .total_vectors = 10_000_000
661660
662661 analyzer = ConfigChangeAnalyzer (
@@ -703,7 +702,7 @@ async def test_handles_zero_vectors(
703702 test_settings : Mock ,
704703 ) -> None :
705704 """Test handling of collection with zero vectors."""
706- checkpoint = await mock_checkpoint_manager .load_checkpoint ()
705+ checkpoint = await mock_checkpoint_manager .load ()
707706 checkpoint .total_vectors = 0
708707
709708 from codeweaver .engine .services .config_analyzer import ConfigChangeAnalyzer
@@ -778,7 +777,7 @@ async def test_breaking_change_provides_recovery_steps(
778777 new_config .dimension = 2048
779778 new_config .datatype = "float32"
780779
781- checkpoint = await mock_checkpoint_manager .load_checkpoint ()
780+ checkpoint = await mock_checkpoint_manager .load ()
782781
783782 # Create fingerprint
784783 from codeweaver .engine .managers .checkpoint_manager import CheckpointSettingsFingerprint
@@ -827,7 +826,7 @@ async def test_transformable_change_provides_strategy(
827826 new_config .dimension = 512 # Reduction
828827 new_config .datatype = "float32"
829828
830- checkpoint = await mock_checkpoint_manager .load_checkpoint ()
829+ checkpoint = await mock_checkpoint_manager .load ()
831830
832831 # Create fingerprint
833832 from codeweaver .engine .managers .checkpoint_manager import CheckpointSettingsFingerprint
@@ -888,7 +887,7 @@ async def test_estimates_scale_with_vector_count(
888887 new_config .dimension = 1024
889888 new_config .datatype = "float32"
890889
891- await mock_checkpoint_manager .load_checkpoint ()
890+ await mock_checkpoint_manager .load ()
892891
893892 # Create fingerprint
894893 from codeweaver .engine .managers .checkpoint_manager import CheckpointSettingsFingerprint
@@ -940,7 +939,7 @@ async def test_no_change_has_zero_estimates(
940939 new_config .dimension = 2048
941940 new_config .datatype = "float32"
942941
943- checkpoint = await mock_checkpoint_manager .load_checkpoint ()
942+ checkpoint = await mock_checkpoint_manager .load ()
944943
945944 # Create fingerprint
946945 from codeweaver .engine .managers .checkpoint_manager import CheckpointSettingsFingerprint
0 commit comments