decrease fps of testing (#1450)

brain-score · Nov 12, 2024 · 98a3d34 · 98a3d34
1 parent 5c4f9e8
commit 98a3d34
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_model_helpers/temporal/activations/test_inferencer.py b/tests/test_model_helpers/temporal/activations/test_inferencer.py
@@ -129,7 +129,7 @@ def test_compute_temporal_context():
 
 @pytest.mark.memory_intense
 @pytest.mark.parametrize("preprocess", ["normal", "downsample"])
-@pytest.mark.parametrize("fps", [1, 40])
+@pytest.mark.parametrize("fps", [1, 10])
 def test_causal_inferencer(preprocess, fps):
     if preprocess == "normal":
         preprocess = dummy_preprocess
@@ -186,4 +186,4 @@ def test_block_inferencer(preprocess, fps):
     manual_compute_values = manual_compute_values.reshape(len(manual_compute_values), -1).astype(output_values.dtype)
     if preprocess == "downsample":
         output_values = output_values[::2]
-    assert np.allclose(output_values, manual_compute_values)
+    assert np.allclose(output_values, manual_compute_values)