From 238b504a189fde5937e9070e9c1c8951600e78ee Mon Sep 17 00:00:00 2001 From: Ben Lonnqvist Date: Tue, 3 Sep 2024 09:25:39 +0200 Subject: [PATCH] fix model loading bug in test --- brainscore_vision/benchmarks/lonnqvist2024/test.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/brainscore_vision/benchmarks/lonnqvist2024/test.py b/brainscore_vision/benchmarks/lonnqvist2024/test.py index 1decfc6aa..803c0bcf4 100644 --- a/brainscore_vision/benchmarks/lonnqvist2024/test.py +++ b/brainscore_vision/benchmarks/lonnqvist2024/test.py @@ -1,7 +1,7 @@ import pytest from pytest import approx -from brainscore_vision import benchmark_registry, load_benchmark +from brainscore_vision import benchmark_registry, load_benchmark, load_model @pytest.mark.parametrize('benchmark', [ @@ -39,7 +39,7 @@ def test_model(self, dataset, expected_raw_score): else: benchmark = f"Lonnqvist2024_{dataset}" benchmark = load_benchmark(benchmark) - model = 'alexnet' + model = load_model('alexnet') score = benchmark(model) raw_score = score.raw # division by ceiling <= 1 should result in higher score @@ -53,7 +53,7 @@ class TestEngineering: ]) def test_accuracy(self, dataset, expected_accuracy): benchmark = load_benchmark(f"Lonnqvist2024_{dataset}") - model = 'alexnet' + model = load_model('alexnet') score = benchmark(model) raw_score = score.raw # division by ceiling <= 1 should result in higher score