added flush

williamFalcon · Jul 26, 2019 · 69894fc · 69894fc
1 parent 914bd9c
commit 69894fc
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/test_tube/log.py b/test_tube/log.py
@@ -170,10 +170,14 @@ def on_exit(self):
 
     def __clean_dir(self):
         files = os.listdir(self.log_dir)
-        print(files)
+
+        if self.rank == 0:
+            return
+
         for f in files:
             if self.process in f:
-                pass
+                self.close()
+                os.remove(f)
 
     def argparse(self, argparser):
         parsed = vars(argparser)