From 4cd7944af7690e4153d28eb3d37f7c35bae106e2 Mon Sep 17 00:00:00 2001 From: Deirdre Kelliher Date: Tue, 18 Jun 2024 11:43:27 -0400 Subject: [PATCH] marking test_look_at_neural_V1 memory intense --- brainscore_vision/model_helpers/generic_plugin_tests.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/brainscore_vision/model_helpers/generic_plugin_tests.py b/brainscore_vision/model_helpers/generic_plugin_tests.py index c31b28128..f52e2d2cd 100644 --- a/brainscore_vision/model_helpers/generic_plugin_tests.py +++ b/brainscore_vision/model_helpers/generic_plugin_tests.py @@ -5,6 +5,7 @@ # noinspection PyUnresolvedReferences from brainscore_core.plugin_management.generic_plugin_tests_helper import pytest_generate_tests from brainscore_vision import BrainModel, load_model +import pytest def test_identifier(identifier: str): @@ -40,7 +41,7 @@ def test_look_at_behavior_probabilities(identifier: str): assert (0 <= predictions.values).all() assert (predictions.values <= 1).all() - +@pytest.mark.memory_intense def test_look_at_neural_V1(identifier: str): model = load_model(identifier) if not ProbeModel().can_start_recording_region(model, recording_target=BrainModel.RecordingTarget.V1):