Fix logging import (#316)

huggingface · Feb 13, 2025 · 272b648 · 272b648
1 parent 7832290
commit 272b648
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 3 deletions.
diff --git a/.gitignore b/.gitignore
@@ -175,4 +175,6 @@ data/
 wandb/
 logs/
 eval_results/
-results/
+results/
+
+.vscode/
diff --git a/src/open_r1/grpo.py b/src/open_r1/grpo.py
@@ -34,7 +34,7 @@
     reasoning_steps_reward,
 )
 from open_r1.utils.callbacks import get_callbacks
-from open_r1.utils.logging import init_wandb_training
+from open_r1.utils.wandb_logging import init_wandb_training
 from trl import GRPOTrainer, ModelConfig, ScriptArguments, TrlParser, get_peft_config
 
 

diff --git a/src/open_r1/sft.py b/src/open_r1/sft.py
@@ -48,7 +48,7 @@
 
 from open_r1.configs import SFTConfig
 from open_r1.utils.callbacks import get_callbacks
-from open_r1.utils.logging import init_wandb_training
+from open_r1.utils.wandb_logging import init_wandb_training
 from trl import (
     ModelConfig,
     ScriptArguments,