diff --git a/aws_xray_sdk/ext/botocore/patch.py b/aws_xray_sdk/ext/botocore/patch.py index 3be9dfdf..f2e53df8 100644 --- a/aws_xray_sdk/ext/botocore/patch.py +++ b/aws_xray_sdk/ext/botocore/patch.py @@ -31,7 +31,9 @@ def _xray_traced_botocore(wrapped, instance, args, kwargs): service = instance._service_model.metadata["endpointPrefix"] if service == 'xray': # skip tracing for SDK built-in sampling pollers - if 'GetSamplingRules' in args or 'GetSamplingTargets' in args: + if ('GetSamplingRules' in args or + 'GetSamplingTargets' in args or + 'PutTraceSegments' in args): return wrapped(*args, **kwargs) return xray_recorder.record_subsegment( wrapped, instance, args, kwargs,