diff --git a/wandb_client.py b/wandb_client.py index 36ad087..c9572a3 100644 --- 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()