update docstring

ServiceNow · Nov 5, 2024 · d9857c5 · d9857c5
1 parent c737e72
commit d9857c5
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/examples/rl_gsm8k/orchestrate_rl.py b/examples/rl_gsm8k/orchestrate_rl.py
@@ -88,6 +88,8 @@ def extract_tape_training_samples(
         agent: MathAgent
         dataset_name: Name of dataset ('train' or 'test')
         tapes_dir: Directory to save processed tapes
+        cfg: Configuration
+        llm_calls: List of LLM calls
 
     Returns:
         Tuple containing: