Skip to content

Commit

Permalink
Fix otel unit tests
Browse files Browse the repository at this point in the history
  • Loading branch information
pquentin committed Feb 22, 2024
1 parent d64be3f commit fcf4127
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions tests/test_otel.py
Original file line number Diff line number Diff line change
Expand Up @@ -36,7 +36,7 @@ def test_minimal_span():
tracer, memory_exporter = setup_tracing()

otel = OpenTelemetry(enabled=True, tracer=tracer)
with otel.span("GET", path_parts={}):
with otel.span("GET", endpoint_id=None, path_parts={}):
pass

spans = memory_exporter.get_finished_spans()
Expand All @@ -52,7 +52,7 @@ def test_detailed_span():
tracer, memory_exporter = setup_tracing()
otel = OpenTelemetry(enabled=True, tracer=tracer)
with otel.span(
"GET", "ml.close_job", path_parts={"job_id": "my-job", "foo": "bar"}
"GET", endpoint_id="ml.close_job", path_parts={"job_id": "my-job", "foo": "bar"}
):
pass

Expand Down

0 comments on commit fcf4127

Please sign in to comment.