Skip to content

Commit 583f12f

Browse files
committed
lint
1 parent da7b9f9 commit 583f12f

File tree

2 files changed

+21
-41
lines changed

2 files changed

+21
-41
lines changed

opentelemetry-sdk/src/opentelemetry/sdk/trace/sampling.py

Lines changed: 16 additions & 20 deletions
Original file line numberDiff line numberDiff line change
@@ -422,26 +422,22 @@ def _get_from_env_or_default() -> Sampler:
422422
rate = 1.0
423423
return _KNOWN_SAMPLERS[trace_sampler_name](rate)
424424
return _KNOWN_SAMPLERS[trace_sampler_name]
425-
else:
426-
try:
427-
trace_sampler_factory = _import_sampler_factory(trace_sampler_name)
428-
sampler_arg = os.getenv(OTEL_TRACES_SAMPLER_ARG, "")
429-
trace_sampler = trace_sampler_factory(sampler_arg)
430-
if not isinstance(trace_sampler, Sampler):
431-
message = (
432-
"Output of traces sampler factory, %s, was not a Sampler object."
433-
% trace_sampler_factory
434-
)
435-
_logger.warning(message)
436-
raise ValueError(message)
437-
return trace_sampler
438-
except Exception as exc:
439-
_logger.warning(
440-
"Using default sampler. Failed to initialize custom sampler, %s: %s",
441-
trace_sampler_name,
442-
exc,
443-
)
444-
return _KNOWN_SAMPLERS["parentbased_always_on"]
425+
try:
426+
trace_sampler_factory = _import_sampler_factory(trace_sampler_name)
427+
sampler_arg = os.getenv(OTEL_TRACES_SAMPLER_ARG, "")
428+
trace_sampler = trace_sampler_factory(sampler_arg)
429+
if not isinstance(trace_sampler, Sampler):
430+
message = f"Output of traces sampler factory, {trace_sampler_factory}, was not a Sampler object."
431+
_logger.warning(message)
432+
raise ValueError(message)
433+
return trace_sampler
434+
except Exception as exc: # pylint: disable=broad-except
435+
_logger.warning(
436+
"Using default sampler. Failed to initialize custom sampler, %s: %s",
437+
trace_sampler_name,
438+
exc,
439+
)
440+
return _KNOWN_SAMPLERS["parentbased_always_on"]
445441

446442

447443
def _get_parent_trace_state(parent_context) -> Optional["TraceState"]:

opentelemetry-sdk/tests/trace/test_trace.py

Lines changed: 5 additions & 21 deletions
Original file line numberDiff line numberDiff line change
@@ -141,10 +141,10 @@ def test_tracer_provider_accepts_concurrent_multi_span_processor(self):
141141

142142
class CustomSampler(sampling.Sampler):
143143
def __init__(self) -> None:
144-
super().__init__()
144+
pass
145145

146146
def get_description(self) -> str:
147-
return super().get_description()
147+
return "CustomSampler"
148148

149149
def should_sample(
150150
self,
@@ -156,15 +156,7 @@ def should_sample(
156156
links,
157157
trace_state,
158158
):
159-
return super().should_sample(
160-
parent_context,
161-
trace_id,
162-
name,
163-
kind,
164-
attributes,
165-
links,
166-
trace_state,
167-
)
159+
return True
168160

169161

170162
class CustomRatioSampler(sampling.TraceIdRatioBased):
@@ -173,7 +165,7 @@ def __init__(self, ratio):
173165
super().__init__(ratio)
174166

175167
def get_description(self) -> str:
176-
return super().get_description()
168+
return "CustomSampler"
177169

178170
def should_sample(
179171
self,
@@ -185,15 +177,7 @@ def should_sample(
185177
links,
186178
trace_state,
187179
):
188-
return super().should_sample(
189-
parent_context,
190-
trace_id,
191-
name,
192-
kind,
193-
attributes,
194-
links,
195-
trace_state,
196-
)
180+
return True
197181

198182

199183
class CustomSamplerFactory:

0 commit comments

Comments
 (0)