#1632 - Fixed output directory clear bug. Added gputil to dev deps.

2023-07-20 10:28:19 +01:00
parent f22681d6b4
commit 3731b2ba13
2 changed files with 3 additions and 1 deletions
--- a/benchmark/primaite_benchmark.py
+++ b/benchmark/primaite_benchmark.py
@@ -32,7 +32,8 @@ _RESULTS_ROOT.mkdir(exist_ok=True, parents=True)

 _OUTPUT_ROOT: Final[Path] = _BENCHMARK_ROOT / "output"
 # Clear and recreate the output directory
-shutil.rmtree(_OUTPUT_ROOT)
+if _OUTPUT_ROOT.exists():
+    shutil.rmtree(_OUTPUT_ROOT)
 _OUTPUT_ROOT.mkdir()

 _TRAINING_CONFIG_PATH = _BENCHMARK_ROOT / "config" / "benchmark_training_config.yaml"