From 644cf78fabf4cf5b25411da4184df5be6686219e Mon Sep 17 00:00:00 2001 From: Alexandre Piche Date: Wed, 22 Jan 2025 22:12:16 +0000 Subject: [PATCH] typo --- examples/rl_gsm8k/orchestrate_rl.py | 1 + 1 file changed, 1 insertion(+) diff --git a/examples/rl_gsm8k/orchestrate_rl.py b/examples/rl_gsm8k/orchestrate_rl.py index 930ad786..49c8c3ad 100644 --- 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