Skip to content

Commit 3dcfd7f

Browse files
committed
refactor(eval): remove unnecessary exports
1 parent 39e915e commit 3dcfd7f

File tree

3 files changed

+5
-8
lines changed

3 files changed

+5
-8
lines changed

docs/evaluation.md

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -247,7 +247,8 @@ Override `get_metric_fns()` to customize metrics:
247247
```python
248248
from functools import partial
249249

250-
from strands_env.eval import Evaluator, compute_pass_at_k
250+
from strands_env.eval import Evaluator
251+
from strands_env.eval.metrics import compute_pass_at_k
251252

252253
class MyEvaluator(Evaluator):
253254
benchmark_name = "my-benchmark"

src/strands_env/eval/__init__.py

Lines changed: 1 addition & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -12,21 +12,16 @@
1212
# See the License for the specific language governing permissions and
1313
# limitations under the License.
1414

15-
from .aime import AIME2024Evaluator, AIME2025Evaluator, AIMEEvaluator
1615
from .evaluator import AsyncEnvFactory, EvalSample, Evaluator
17-
from .metrics import MetricFn, compute_pass_at_k
16+
from .metrics import MetricFn
1817
from .registry import get_benchmark, list_benchmarks, register_eval
1918

2019
__all__ = [
21-
"AIME2024Evaluator",
22-
"AIME2025Evaluator",
23-
"AIMEEvaluator",
2420
"AsyncEnvFactory",
2521
"EvalSample",
2622
"Evaluator",
2723
"MetricFn",
2824
"get_benchmark",
2925
"list_benchmarks",
30-
"compute_pass_at_k",
3126
"register_eval",
3227
]

tests/unit/test_eval.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,8 @@
55
import pytest
66

77
from strands_env.core import Action, Environment, Observation, RewardResult, StepResult, TaskContext
8-
from strands_env.eval import AIME2024Evaluator, EvalSample, Evaluator
8+
from strands_env.eval import EvalSample, Evaluator
9+
from strands_env.eval.aime import AIME2024Evaluator
910
from strands_env.eval.metrics import compute_pass_at_k
1011

1112
# ---------------------------------------------------------------------------

0 commit comments

Comments
 (0)