Address review comments

aristanetworks · Sep 25, 2024 · a29d330 · a29d330
1 parent 6484f72
commit a29d330
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/tests/benchmark/test_anta.py b/tests/benchmark/test_anta.py
@@ -41,7 +41,8 @@ def bench() -> ResultManager:
         asyncio.run(main(manager, inventory, catalog, dry_run=True))
         return manager
 
-    manager = benchmark(lambda: bench())  # pylint: disable=unnecessary-lambda
+    manager = benchmark(bench)
+
     logging.disable(logging.NOTSET)
     if len(manager.results) != 0:
         pytest.fail("ANTA Dry-Run mode should not return any result", pytrace=False)
@@ -75,7 +76,8 @@ def bench() -> ResultManager:
         asyncio.run(main(manager, inventory, catalog))
         return manager
 
-    manager = benchmark(lambda: bench())  # pylint: disable=unnecessary-lambda
+    manager = benchmark(bench)
+
     logging.disable(logging.NOTSET)
 
     if len(catalog.tests) * len(inventory) != len(manager.results):