Update file logger.py

logger.py

@@ -26,5 +26,5 @@ class Wandb:
         wandb.log(args)
 
 
-    def __del__(self):
+    def __exit__(self):
         wandb.finish()