Skip to content

Commit ccd22c1

Browse files
author
Dylan Huang
committed
rename to match official capitalization
1 parent 034fde7 commit ccd22c1

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

eval_protocol/logging/elasticsearch_direct_http_handler.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,7 @@
1111
from .elasticsearch_client import ElasticsearchClient, ElasticsearchConfig as ESConfig
1212

1313

14-
class ElasticSearchDirectHttpHandler(logging.Handler):
14+
class ElasticsearchDirectHttpHandler(logging.Handler):
1515
def __init__(self, elasticsearch_config: ElasticSearchConfig) -> None:
1616
super().__init__()
1717
self.config = ESConfig(
@@ -55,7 +55,7 @@ def _get_rollout_id(self, record: logging.LogRecord) -> str:
5555
rollout_id = os.getenv("EP_ROLLOUT_ID")
5656
if rollout_id is None:
5757
raise ValueError(
58-
"EP_ROLLOUT_ID environment variable is not set but needed for ElasticSearchDirectHttpHandler"
58+
"EP_ROLLOUT_ID environment variable is not set but needed for ElasticsearchDirectHttpHandler"
5959
)
6060
return rollout_id
6161

tests/logging/test_elasticsearch_direct_http_handler.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33
import time
44
import pytest
55

6-
from eval_protocol.logging.elasticsearch_direct_http_handler import ElasticSearchDirectHttpHandler
6+
from eval_protocol.logging.elasticsearch_direct_http_handler import ElasticsearchDirectHttpHandler
77
from eval_protocol.logging.elasticsearch_client import ElasticsearchClient, ElasticsearchConfig as ESConfig
88
from eval_protocol.pytest.elasticsearch_setup import ElasticsearchSetup
99
from eval_protocol.types.remote_rollout_processor import ElasticSearchConfig
@@ -43,7 +43,7 @@ def elasticsearch_handler(elasticsearch_config: ElasticSearchConfig, rollout_id:
4343
"""Create and configure ElasticsearchDirectHttpHandler."""
4444
# Use a unique test-specific index name with timestamp
4545

46-
handler = ElasticSearchDirectHttpHandler(elasticsearch_config)
46+
handler = ElasticsearchDirectHttpHandler(elasticsearch_config)
4747

4848
# Set a specific log level
4949
handler.setLevel(logging.INFO)

0 commit comments

Comments
 (0)