fuck wandb

Author: alexdremov

ml_research_tools/exp/wandb_downloader_tool.py

@@ -280,6 +280,7 @@ def process_run(self, run: WandbRun, output_dir: str) -> None:
             output_dir: Directory where the log file will be saved
         """
         run.load_full_data()
+        run.load()
 
         # Sanitize the run name for use in filenames
         sanitized_name = self.sanitize_filename(run.name)
@@ -320,7 +321,7 @@ def process_run(self, run: WandbRun, output_dir: str) -> None:
 
         # Extract the history of the run as a dataframe
         try:
-            history = run.history(pandas=True)
+            history = run.history(pandas=True, samples=None)
         except Exception as e:
             self.logger.warning(f"Failed to retrieve history for run {run.name}/{run.id}: {e}")
             return