Del objects in runner

aristanetworks · Aug 28, 2024 · ccb02dc · ccb02dc
1 parent 0c88cf6
commit ccb02dc
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/anta/runner.py b/anta/runner.py
@@ -315,6 +315,10 @@ async def main(  # noqa: PLR0913
             )
 
         tests_generator = generate_tests(selected_tests)
+        total_tests = catalog.final_tests_count
+
+        # Cleanup no longer needed objects before running the tests
+        del inventory, catalog, selected_tests
 
     if dry_run:
         logger.info("Dry-run mode, exiting before running the tests.")
@@ -323,7 +327,7 @@ async def main(  # noqa: PLR0913
         return
 
     if AntaTest.progress is not None:
-        AntaTest.nrfu_task = AntaTest.progress.add_task("Running NRFU Tests...", total=catalog.final_tests_count)
+        AntaTest.nrfu_task = AntaTest.progress.add_task("Running NRFU Tests...", total=total_tests)
 
     with Catchtime(logger=logger, message="Running ANTA tests"):
         async for result in run_tests(tests_generator, limit=max_concurrency):