Merge branch 'main' into XianzheMa/bug/fix-505

modyn/tests/trainer_server/internal/trainer/remote_downsamplers/test_craig_remote_downsampling.py

@@ -21,6 +21,7 @@ def get_sampler_config(modyn_config, balance=False):
         "selection_batch": 64,
         "greedy": "NaiveGreedy",
         "full_grad_approximation": "LastLayerWithEmbedding",
+        "ratio_max": 100,
     }
     return 0, 0, 0, params_from_selector, modyn_config.model_dump(by_alias=True), per_sample_loss_fct, "cpu"
 
@@ -354,6 +355,7 @@ def test_matching_results_with_deepcore(dummy_system_config: ModynConfig):
             "selection_batch": 64,
             "greedy": "NaiveGreedy",
             "full_grad_approximation": "LastLayerWithEmbedding",
+            "ratio_max": 100,
         },
         dummy_system_config.model_dump(by_alias=True),
         BCEWithLogitsLoss(reduction="none"),
@@ -415,6 +417,7 @@ def test_matching_results_with_deepcore_permutation(dummy_system_config: ModynCo
             "selection_batch": 64,
             "greedy": "NaiveGreedy",
             "full_grad_approximation": "LastLayerWithEmbedding",
+            "ratio_max": 100,
         },
         dummy_system_config.model_dump(by_alias=True),
         BCEWithLogitsLoss(reduction="none"),
@@ -480,6 +483,7 @@ def test_matching_results_with_deepcore_permutation_fancy_ids(dummy_system_confi
             "selection_batch": 64,
             "greedy": "NaiveGreedy",
             "full_grad_approximation": "LastLayerWithEmbedding",
+            "ratio_max": 100,
         },
         dummy_system_config.model_dump(by_alias=True),
         BCEWithLogitsLoss(reduction="none"),

modyn/tests/trainer_server/internal/trainer/remote_downsamplers/test_remote_gradmatch_downsampling_strategy.py

@@ -20,6 +20,7 @@ def get_sampler_config(modyn_config: ModynConfig, balance=False):
         "args": {},
         "balance": balance,
         "full_grad_approximation": "LastLayerWithEmbedding",
+        "ratio_max": 100,
     }
     return 0, 0, 0, params_from_selector, modyn_config.model_dump(by_alias=True), per_sample_loss_fct, "cpu"