one more cleaning up of check_deterministic

RagnarGrootKoerkamp · Nov 14, 2024 · 69dbdbf · 69dbdbf
1 parent 7df2adb
commit 69dbdbf
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/bin/generate.py b/bin/generate.py
@@ -825,6 +825,8 @@ def check_deterministic(force=False):
                     new_seed = (t.seed + 1 + run) % (2**31)
                     result = t.generator.run(bar, tmp, tmp_infile.stem, new_seed, t.config.retries)
                     if not result.status:
+                        # clean up
+                        shutil.rmtree(tmp)
                         return
 
                     # Now check that the source and target are different.