typo

ServiceNow · Jan 22, 2025 · 644cf78 · 644cf78
1 parent 7ffe858
commit 644cf78
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/examples/rl_gsm8k/orchestrate_rl.py b/examples/rl_gsm8k/orchestrate_rl.py
@@ -19,6 +19,7 @@
 from termcolor import colored
 from tqdm import tqdm
 
+import wandb
 from tapeagents.agent import Agent
 from tapeagents.core import LLMCall, LLMOutputParsingFailureAction, StepMetadata, TrainingText
 from tapeagents.finetune.data import MASKED_TOKEN_ID