From 98a3d3444e2e5da6187ca1075c714483ea3d2496 Mon Sep 17 00:00:00 2001 From: Sam Winebrake <85908068+samwinebrake@users.noreply.github.com> Date: Tue, 12 Nov 2024 16:00:22 -0500 Subject: [PATCH] decrease fps of testing (#1450) --- .../temporal/activations/test_inferencer.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/test_model_helpers/temporal/activations/test_inferencer.py b/tests/test_model_helpers/temporal/activations/test_inferencer.py index 06e8a775d..d9df0b5d4 100644 --- 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) \ No newline at end of file + assert np.allclose(output_values, manual_compute_values)