Skip to content

Commit f4d80ec

Browse files
author
Dylan Huang
committed
Added Status model import to eval_protocol and included it in the __all__ exports for better accessibility.
1 parent dcaf023 commit f4d80ec

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

eval_protocol/__init__.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,7 @@
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, InputMetadata
36+
from .models import EvaluateResult, Message, MetricResult, EvaluationRow, InputMetadata, Status
3737
from .playback_policy import PlaybackPolicyBase
3838
from .resources import create_llm_resource
3939
from .reward_function import RewardFunction
@@ -73,6 +73,7 @@
7373
warnings.filterwarnings("default", category=DeprecationWarning, module="eval_protocol")
7474

7575
__all__ = [
76+
"Status",
7677
"RemoteRolloutProcessor",
7778
"InputMetadata",
7879
"EvaluationRow",

0 commit comments

Comments
 (0)