From d9857c5d1d3b3773ac604037146bce8245c04f86 Mon Sep 17 00:00:00 2001
From: Alexandre Piche <alexandrelpiche@gmail.com>
Date: Tue, 5 Nov 2024 03:39:42 +0000
Subject: [PATCH] update docstring

---
 examples/rl_gsm8k/orchestrate_rl.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/examples/rl_gsm8k/orchestrate_rl.py b/examples/rl_gsm8k/orchestrate_rl.py
index 8ae4ff7..79a0f38 100644
--- 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: