potential thread safety fix

- related: #548
tqdm · Aug 27, 2019 · a347706 · a347706 · casperdcl · Aug 28, 2019
1 parent 96cd73e
commit a347706
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/tqdm/std.py b/tqdm/std.py
@@ -1022,9 +1022,13 @@ def __len__(self):
     def __enter__(self):
         return self
 
-    def __exit__(self, *exc):
-        self.close()
-        return False
+    def __exit__(self, exc_type, exc_value, traceback):
+        try:
+            self.close()
+        except AttributeError:
+            # maybe eager thread cleanup upon external error
+            if exc_type is None:
+                raise
 
     def __del__(self):
         self.close()