Skip to content

Commit 712a37d

Browse files
committed
Merge branch 'derekx/fireworks-tracing' of https://github.com/eval-protocol/python-sdk into derekx/fireworks-tracing
2 parents 91f2378 + f4527eb commit 712a37d

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

eval_protocol/pytest/remote_rollout_processor.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -64,6 +64,10 @@ def __init__(
6464
self._model_base_url = model_base_url
6565
if os.getenv("EP_REMOTE_ROLLOUT_PROCESSOR_BASE_URL"):
6666
self._remote_base_url = os.getenv("EP_REMOTE_ROLLOUT_PROCESSOR_BASE_URL")
67+
self._model_base_url = model_base_url
68+
_ep_model_base_url = os.getenv("EP_MODEL_BASE_URL")
69+
if _ep_model_base_url:
70+
self._model_base_url = _ep_model_base_url
6771
self._poll_interval = poll_interval
6872
self._timeout_seconds = timeout_seconds
6973
self._output_data_loader = output_data_loader or _default_output_data_loader

0 commit comments

Comments
 (0)