update

camel-ai · Jan 30, 2025 · 7169cea · 7169cea
1 parent f46e3c6
commit 7169cea
Show file tree

Hide file tree

Showing 3 changed files with 3,363 additions and 2 deletions.
diff --git a/camel/datagen/star/star_pipeline.py b/camel/datagen/star/star_pipeline.py
@@ -580,6 +580,7 @@ def process_problem(
         solution_text = problem.get("solution", "")
         current_trace = self.generate_reasoning_trace(problem_text)
         improvement_history = []
+        scores = {}
 
         # Create batches for parallel evaluation
         batch_problems = []

diff --git a/examples/datagen/star/download_math500_dataset.py b/examples/datagen/star/download_math500_dataset.py
@@ -23,8 +23,8 @@ def download_huggingface_dataset():
         # Load the dataset using the datasets library
         dataset = load_dataset("HuggingFaceH4/MATH-500", split="test")
 
-        # Get the first 20 items
-        data = dataset.select(range(20))
+        # Get the first 500 items
+        data = dataset.select(range(500))
 
         # Print the first item to see its structure
         print("Dataset structure:")