fix

Author: Dylan Huang

eval_protocol/__init__.py

@@ -34,7 +34,7 @@
 from .pytest.parameterize import DefaultParameterIdGenerator
 from .log_utils.elasticsearch_direct_http_handler import ElasticsearchDirectHttpHandler
 from .log_utils.rollout_id_filter import RolloutIdFilter
-from .log_utils.setup import setup_rollout_logging_for_elasticsearch_handler
+from .log_utils.util import setup_rollout_logging_for_elasticsearch_handler
 
 from .types.remote_rollout_processor import (
     InitRequest,

eval_protocol/log_utils/util.py

@@ -1,6 +1,6 @@
 import os
 from eval_protocol.types.remote_rollout_processor import ElasticsearchConfig
-from tests.logging.test_elasticsearch_direct_http_handler import ElasticsearchDirectHttpHandler
+from .elasticsearch_direct_http_handler import ElasticsearchDirectHttpHandler
 
 
 def setup_rollout_logging_for_elasticsearch_handler(