output log fix (#213)

wandb · Sep 16, 2020 · cea3358 · cea3358
1 parent 215e853
commit cea3358
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/wandb/sdk/wandb_run.py b/wandb/sdk/wandb_run.py
@@ -1521,4 +1521,8 @@ def write(self, *args, **kargs):
             self.lock.release()
 
     def close(self):
-        self.f.close()
+        self.lock.acquire()  # wait for pending writes
+        try:
+            self.f.close()
+        finally:
+            self.lock.release()
diff --git a/wandb/sdk_py27/wandb_run.py b/wandb/sdk_py27/wandb_run.py
@@ -1521,4 +1521,8 @@ def write(self, *args, **kargs):
             self.lock.release()
 
     def close(self):
-        self.f.close()
+        self.lock.acquire()  # wait for pending writes
+        try:
+            self.f.close()
+        finally:
+            self.lock.release()