From 24bad9bbaa6ad7560c8ff900f5452b75c744a993 Mon Sep 17 00:00:00 2001 From: Songki Choi Date: Mon, 19 Feb 2024 11:07:00 +0900 Subject: [PATCH] Fix missing fixture in perf test --- tests/perf/test_classification.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/perf/test_classification.py b/tests/perf/test_classification.py index 820d644ae40..9397dc5413e 100644 --- a/tests/perf/test_classification.py +++ b/tests/perf/test_classification.py @@ -52,7 +52,7 @@ class TestPerfSingleLabelClassification: @pytest.mark.parametrize("fxt_model_id", MODEL_TEMPLATES, ids=MODEL_IDS, indirect=True) @pytest.mark.parametrize("fxt_benchmark", BENCHMARK_CONFIGS.items(), ids=BENCHMARK_CONFIGS.keys(), indirect=True) - def test_accuracy(self, fxt_model_id: str, fxt_benchmark: OTXBenchmark): + def test_accuracy(self, fxt_model_id: str, fxt_benchmark: OTXBenchmark, fxt_check_benchmark_result: Callable): """Benchmark accruacy metrics.""" result = fxt_benchmark.run( model_id=fxt_model_id, @@ -301,7 +301,7 @@ def test_accuracy(self, fxt_model_id: str, fxt_benchmark: OTXBenchmark, fxt_chec @pytest.mark.parametrize("fxt_model_id", MODEL_TEMPLATES, ids=MODEL_IDS, indirect=True) @pytest.mark.parametrize("fxt_benchmark", BENCHMARK_CONFIGS.items(), ids=BENCHMARK_CONFIGS.keys(), indirect=True) - def test_speed(self, fxt_model_id: str, fxt_benchmark: OTXBenchmark, fxt_check_benchmark_results: Callable): + def test_speed(self, fxt_model_id: str, fxt_benchmark: OTXBenchmark, fxt_check_benchmark_result: Callable): """Benchmark train time per iter / infer time per image.""" fxt_benchmark.track_resources = True result = fxt_benchmark.run(