suprress exception in wandb.finish

replicate · Sep 4, 2024 · 1d7d755 · 1d7d755
1 parent 6aee89b
commit 1d7d755
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/wandb_client.py b/wandb_client.py
@@ -1,5 +1,6 @@
 from pathlib import Path
 from typing import Any, Sequence
+from contextlib import suppress
 import wandb
 from wandb.sdk.wandb_settings import Settings
 
@@ -40,4 +41,5 @@ def save_weights(self, lora_path: Path):
         wandb.save(lora_path)
 
     def finish(self):
-        wandb.finish()
+        with suppress(Exception):
+            wandb.finish()