refactor(envs): rename SimpleMathEnv to CalculatorEnv

- Rename simple_math_env/ directory to calculator/
- Rename SimpleMathEnv class to CalculatorEnv
- Update environments/__init__.py exports

Co-Authored-By: Claude Opus 4.5 <[email protected]>

Author: Lawhy

examples/aime_eval.py

@@ -33,7 +33,7 @@
 import click
 from common import ModelConfig, SamplingParams
 
-from strands_env.environments.simple_math_env import SimpleMathEnv
+from strands_env.environments.calculator.env import CalculatorEnv
 from strands_env.eval import AIMEEvaluator
 from strands_env.rewards.math_reward import MathRewardFunction
 
@@ -49,7 +49,7 @@ async def run_eval(
     reward_fn = MathRewardFunction()
 
     async def env_factory(_):
-        env = SimpleMathEnv(model_factory=model_factory, reward_fn=reward_fn, verbose=False)
+        env = CalculatorEnv(model_factory=model_factory, reward_fn=reward_fn, verbose=False)
         env.system_prompt = None
         env.get_tools = lambda: []
         return env

examples/math_env.py

@@ -31,7 +31,7 @@
 from common import ModelConfig, SamplingParams
 
 from strands_env.core.types import Action, TaskContext
-from strands_env.environments.simple_math_env import SimpleMathEnv
+from strands_env.environments.calculator.env import CalculatorEnv
 from strands_env.rewards.math_reward import MathRewardFunction
 
 MATH_PROBLEMS = [
@@ -43,7 +43,7 @@
 
 async def run_math_env(config: ModelConfig, verbose: bool) -> None:
     model_factory = config.create_factory()
-    env = SimpleMathEnv(model_factory=model_factory, reward_fn=MathRewardFunction(), verbose=verbose)
+    env = CalculatorEnv(model_factory=model_factory, reward_fn=MathRewardFunction(), verbose=verbose)
 
     for question, ground_truth in MATH_PROBLEMS:
         click.echo(f"\n{'=' * 60}")

src/strands_env/environments/__init__.py

@@ -14,6 +14,6 @@
 
 """Environments for strands-env."""
 
-from .simple_math_env import SimpleMathEnv
+from .calculator import CalculatorEnv
 
-__all__ = ["SimpleMathEnv"]
+__all__ = ["CalculatorEnv"]

…environments/simple_math_env/__init__.py …_env/environments/calculator/__init__.pysrc/strands_env/environments/simple_math_env/__init__.py renamed to src/strands_env/environments/calculator/__init__.py src/strands_env/environments/simple_math_env/__init__.py renamed to src/strands_env/environments/calculator/__init__.py

@@ -14,6 +14,6 @@
 
 """Simple math environment with a calculator tool."""
 
-from .env import SimpleMathEnv
+from .env import CalculatorEnv
 
-__all__ = ["SimpleMathEnv"]
+__all__ = ["CalculatorEnv"]

…_env/environments/simple_math_env/env.py …rands_env/environments/calculator/env.pysrc/strands_env/environments/simple_math_env/env.py renamed to src/strands_env/environments/calculator/env.py src/strands_env/environments/simple_math_env/env.py renamed to src/strands_env/environments/calculator/env.py

@@ -21,7 +21,7 @@
 from strands_env.core.environment import Environment
 
 
-class SimpleMathEnv(Environment):
+class CalculatorEnv(Environment):
     """Simple math environment using a calculator tool."""
 
     default_system_prompt_path = Path(__file__).parent / "system_prompt.md"