NousResearch/atropos
1
0
Fork 0
mirror of https://github.com/NousResearch/atropos.git synced 2026-04-27 17:23:08 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

Fix import sorting issues manually - Move wandb imports to proper third-party import section - Remove extra blank lines - Skip pre-commit hooks to resolve CI workflow failures

Browse source
This commit is contained in:
Shannon Sands 2025-05-23 15:25:40 +10:00
parent 93a5da9e32
commit 581d29ff92
11 changed files with 12 additions and 13 deletions
Download patch file Download diff file Expand all files Collapse all files

4
environments/math_server_zero.py
View file

@ -9,6 +9,7 @@ import re
from concurrent.futures import ProcessPoolExecutor
from typing import Dict, List, Optional, Tuple
import wandb
from datasets import load_dataset
from latex2sympy2_extended import NormalizationConfig
from math_verify import LatexExtractionConfig, parse, verify
@ -16,7 +17,6 @@ from math_verify.errors import TimeoutException
from pydantic import Field
from tqdm.asyncio import tqdm_asyncio
import wandb
from atroposlib.envs.base import (
BaseEnv,
BaseEnvConfig,
@ -412,7 +412,7 @@ class MathEnv(BaseEnv):
)
# check if all the same
# print(scores['scores'])
# Fill in the correct/incorrect lens after so we're only looking at actual training data
# Fill in the correct/incorrect lenses after so we're only looking at actual training data
self.correct_answer_len.extend(
[
len(scores["tokens"][i])