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

environments/community/philosophical_rlaif_env.py

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

environments/community/ufc_prediction_env/ufc_server.py

@@ -88,21 +88,21 @@ class UFCEnv(BaseEnv):
                     return f"{name}: (No stats available)"
                 return (
                     f"Name: {name}\n"
-                    f"Wins: {stats.get('wins','?')}  "
+                    f"Wins: {stats.get('wins', '?')}  "
-                    f"Losses: {stats.get('losses','?')}  "
+                    f"Losses: {stats.get('losses', '?')}  "
-                    f"Age: {stats.get('age','?')}\n"
+                    f"Age: {stats.get('age', '?')}\n"
-                    f"Height: {stats.get('height','?')} cm  "
+                    f"Height: {stats.get('height', '?')} cm  "
-                    f"Weight: {stats.get('weight','?')} kg  "
+                    f"Weight: {stats.get('weight', '?')} kg  "
-                    f"Reach: {stats.get('reach','?')} cm  "
+                    f"Reach: {stats.get('reach', '?')} cm  "
-                    f"Stance: {stats.get('stance','?')}\n"
+                    f"Stance: {stats.get('stance', '?')}\n"
-                    f"SLpM: {stats.get('SLpM','?')}  "
+                    f"SLpM: {stats.get('SLpM', '?')}  "
-                    f"Sig Str Acc: {stats.get('sig_str_acc','?')}  "
+                    f"Sig Str Acc: {stats.get('sig_str_acc', '?')}  "
-                    f"SApM: {stats.get('SApM','?')}  "
+                    f"SApM: {stats.get('SApM', '?')}  "
-                    f"Str Def: {stats.get('str_def','?')}\n"
+                    f"Str Def: {stats.get('str_def', '?')}\n"
-                    f"TD Avg: {stats.get('td_avg','?')}  "
+                    f"TD Avg: {stats.get('td_avg', '?')}  "
-                    f"TD Acc: {stats.get('td_acc','?')}  "
+                    f"TD Acc: {stats.get('td_acc', '?')}  "
-                    f"TD Def: {stats.get('td_def','?')}  "
+                    f"TD Def: {stats.get('td_def', '?')}  "
-                    f"Sub Avg: {stats.get('sub_avg','?')}\n"
+                    f"Sub Avg: {stats.get('sub_avg', '?')}\n"
                 )
 
             prompt_text = (

environments/math_server.py

@@ -5,6 +5,7 @@ from concurrent.futures import ProcessPoolExecutor
 from difflib import SequenceMatcher
 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
@@ -12,7 +13,6 @@ from math_verify.errors import TimeoutException
 from pydantic import Field
 from tqdm.asyncio import tqdm_asyncio
 
-import wandb
 from atroposlib.envs.base import (
     APIServerConfig,
     BaseEnv,

environments/math_server_zero.py

@@ -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,

environments/tool_calling_server.py

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