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

Fix import sorting issues in environment files - Move wandb imports to third-party section in philosophical_rlaif_env.py, math_server.py, math_server_zero.py, and tool_calling_server.py - Add missing spaces after commas in ufc_server.py stats.get() calls - Manual fixes to resolve CI linting failures

Browse source
This commit is contained in:
Shannon Sands 2025-05-23 15:58:16 +10:00
parent 285b2bb2b7
commit 8aff7cb3c5
5 changed files with 19 additions and 19 deletions
Download patch file Download diff file Expand all files Collapse all files

2
environments/community/philosophical_rlaif_env.py
View file

@ -3,10 +3,10 @@ import random
import re # Added import for regular expressions import re # Added import for regular expressions
from typing import Dict, List, Optional, Tuple from typing import Dict, List, Optional, Tuple
import wandb # type: ignore
from datasets import load_dataset # type: ignore from datasets import load_dataset # type: ignore
from pydantic import Field from pydantic import Field
import wandb # type: ignore
from atroposlib.envs.base import BaseEnvConfig # Ensure BaseEnvConfig is imported from atroposlib.envs.base import BaseEnvConfig # Ensure BaseEnvConfig is imported
from atroposlib.envs.base import ( from atroposlib.envs.base import (
APIServerConfig, APIServerConfig,

2
environments/math_server.py
View file

@ -5,6 +5,7 @@ from concurrent.futures import ProcessPoolExecutor
from difflib import SequenceMatcher from difflib import SequenceMatcher
from typing import Dict, List, Optional, Tuple from typing import Dict, List, Optional, Tuple
import wandb
from datasets import load_dataset from datasets import load_dataset
from latex2sympy2_extended import NormalizationConfig from latex2sympy2_extended import NormalizationConfig
from math_verify import LatexExtractionConfig, parse, verify from math_verify import LatexExtractionConfig, parse, verify
@ -12,7 +13,6 @@ from math_verify.errors import TimeoutException
from pydantic import Field from pydantic import Field
from tqdm.asyncio import tqdm_asyncio from tqdm.asyncio import tqdm_asyncio
import wandb
from atroposlib.envs.base import ( from atroposlib.envs.base import (
APIServerConfig, APIServerConfig,
BaseEnv, BaseEnv,

2
environments/math_server_zero.py
View file

@ -9,6 +9,7 @@ import re
from concurrent.futures import ProcessPoolExecutor from concurrent.futures import ProcessPoolExecutor
from typing import Dict, List, Optional, Tuple from typing import Dict, List, Optional, Tuple
import wandb
from datasets import load_dataset from datasets import load_dataset
from latex2sympy2_extended import NormalizationConfig from latex2sympy2_extended import NormalizationConfig
from math_verify import LatexExtractionConfig, parse, verify from math_verify import LatexExtractionConfig, parse, verify
@ -16,7 +17,6 @@ from math_verify.errors import TimeoutException
from pydantic import Field from pydantic import Field
from tqdm.asyncio import tqdm_asyncio from tqdm.asyncio import tqdm_asyncio
import wandb
from atroposlib.envs.base import ( from atroposlib.envs.base import (
BaseEnv, BaseEnv,
BaseEnvConfig, BaseEnvConfig,

2
environments/tool_calling_server.py
View file

@ -3,10 +3,10 @@ import random
import re import re
from typing import Dict, List, Optional, Tuple, Union from typing import Dict, List, Optional, Tuple, Union
import wandb
from datasets import load_dataset from datasets import load_dataset
from tqdm.asyncio import tqdm_asyncio from tqdm.asyncio import tqdm_asyncio
import wandb
from atroposlib.envs.base import ( from atroposlib.envs.base import (
APIServerConfig, APIServerConfig,
BaseEnv, BaseEnv,