test: fix benchmark

aristanetworks · Dec 29, 2024 · c0cfda9 · c0cfda9
1 parent 7dd36f0
commit c0cfda9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/benchmark/utils.py b/tests/benchmark/utils.py
@@ -90,7 +90,7 @@ def import_test_modules() -> Generator[ModuleType, None, None]:
             for test_data in module.DATA:
                 test = test_data["test"]
                 result_overwrite = AntaTest.Input.ResultOverwrite(custom_field=test_data["name"])
-                if test_data["inputs"] is None:
+                if "inputs" not in test_data or test_data["inputs"] is None:
                     inputs = test.Input(result_overwrite=result_overwrite)
                 else:
                     inputs = test.Input(**test_data["inputs"], result_overwrite=result_overwrite)