From a7db4c77bbec06649653a9314289c39516933d40 Mon Sep 17 00:00:00 2001
From: Tibo De Peuter <tibo.depeuter@ugent.be>
Date: Thu, 7 May 2026 22:50:38 +0200
Subject: [PATCH] refactor(log): extract w&b file sync

---
 src/experiment_logger/simple_logger.py  |  4 ++++
 src/experiment_logger/unified_logger.py | 15 +++++++++++++++
 2 files changed, 19 insertions(+)

diff --git a/src/experiment_logger/simple_logger.py b/src/experiment_logger/simple_logger.py
index 7e4a816..71844dc 100644
--- a/src/experiment_logger/simple_logger.py
+++ b/src/experiment_logger/simple_logger.py
@@ -71,6 +71,10 @@ class SimpleLogger:
     def save_final_model(self, params: Any, metadata: Optional[Dict[str, Any]] = None):
         print("[SAVE] Final model would be saved (SimpleLogger: No-Op)")
 
+    def sync_file(self, path: Any):
+        """No-op for SimpleLogger."""
+        pass
+
     def finish(self):
         print(f"[FINISH] SimpleLogger finished for run: {self.run_name}")
 
diff --git a/src/experiment_logger/unified_logger.py b/src/experiment_logger/unified_logger.py
index f37136b..e308d12 100644
--- a/src/experiment_logger/unified_logger.py
+++ b/src/experiment_logger/unified_logger.py
@@ -432,6 +432,21 @@ class UnifiedLogger:
         except Exception as e:
             self.error(f"Error saving final model: {e}")
 
+    def sync_file(self, path: Path) -> None:
+        """Upload a file to W&B if tracking is enabled.
+
+        Best-effort: logs a warning on failure, never raises.
+        """
+        if self.wandb_run is None:
+            return
+        try:
+            import wandb
+
+            # "Simple sync" behavior: wandb will copy this file into the run.
+            wandb.save(str(path), base_path=str(path.parent))
+        except Exception as e:
+            self.warning(f"Failed to sync file to W&B: {e}")
+
     def finish(self):
         """Finalize logging and cleanup."""
         # Flush remaining metrics