1
Fork 0

fix: mypy complaints

This commit is contained in:
Tibo De Peuter 2026-04-08 23:37:01 +02:00
parent f6dc9c8e7f
commit 75c6d5bacd
Signed by: tdpeuter
SSH key fingerprint: SHA256:u/h/LVoqKF1Iz02uOyxe6hcjmoZASCGV2HM0TG9ZMoU
7 changed files with 26 additions and 23 deletions

View file

@ -224,7 +224,7 @@ class UnifiedLogger:
self._log_to_stdout(metrics_with_metadata)
# Log to WandB
if self.wandb_available:
if self.wandb_run is not None:
try:
self.wandb_run.log(metrics, step=step, commit=commit)
except Exception as e:
@ -306,7 +306,7 @@ class UnifiedLogger:
self.info(f"Checkpoint saved: {checkpoint_path}")
# Log to WandB as artifact
if self.wandb_available:
if self.wandb_run is not None:
try:
import wandb
@ -342,7 +342,7 @@ class UnifiedLogger:
self.info(f"Final model saved: {final_model_path}")
# Log to WandB
if self.wandb_available:
if self.wandb_run is not None:
try:
import wandb