Skip to content

Commit f9c6f1b

Browse files
author
Dylan Huang
authored
add more to top level import (#221)
1 parent 3b64b31 commit f9c6f1b

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

eval_protocol/__init__.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -33,13 +33,13 @@
3333
_FIREWORKS_AVAILABLE = False
3434
# Import submodules to make them available via eval_protocol.rewards, etc.
3535
from . import mcp, rewards
36-
from .models import EvaluateResult, Message, MetricResult, EvaluationRow
36+
from .models import EvaluateResult, Message, MetricResult, EvaluationRow, InputMetadata
3737
from .playback_policy import PlaybackPolicyBase
3838
from .resources import create_llm_resource
3939
from .reward_function import RewardFunction
4040
from .typed_interface import reward_function
4141
from .quickstart import aha_judge, multi_turn_assistant_to_ground_truth, assistant_to_ground_truth
42-
from .pytest import evaluation_test, SingleTurnRolloutProcessor
42+
from .pytest import evaluation_test, SingleTurnRolloutProcessor, RemoteRolloutProcessor
4343
from .pytest.parameterize import DefaultParameterIdGenerator
4444

4545
try:
@@ -73,6 +73,9 @@
7373
warnings.filterwarnings("default", category=DeprecationWarning, module="eval_protocol")
7474

7575
__all__ = [
76+
"RemoteRolloutProcessor",
77+
"InputMetadata",
78+
"EvaluationRow",
7679
"DefaultParameterIdGenerator",
7780
"DynamicDataLoader",
7881
"InlineDataLoader",

0 commit comments

Comments
 (0)