fix test

eth-easl · Sep 6, 2024 · 4524018 · 4524018
1 parent b514384
commit 4524018
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/modyn/tests/trainer_server/internal/trainer/remote_downsamplers/test_get_tensor_subset.py b/modyn/tests/trainer_server/internal/trainer/remote_downsamplers/test_get_tensor_subset.py
@@ -61,7 +61,7 @@ def test_get_tensors_subset_invalid_index():
     # Test when selected_indexes contains an invalid index
     selected_indexes.append(999)
 
-    with pytest.raises(ValueError):
+    with pytest.raises(KeyError):
         get_tensors_subset(selected_indexes, data, target, sample_ids)