hiro/ray
1
0
Fork 0
mirror of https://github.com/vale981/ray synced 2025-03-06 02:21:39 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions

[client] remove ray_trace_ctx from kwargs if tracing disabled (#18926)

Browse source
This commit is contained in:
Chris K. W 2021-09-28 09:47:43 -07:00 • committed by Jiajun Yao
parent 4da70fbf06
commit 08a7bef9e0
1 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
python/ray/util/tracing/tracing_helper.py
View file

@ -366,7 +366,8 @@ def _tracing_actor_creation(method):
# If tracing feature flag is not on, perform a no-op
if not is_tracing_enabled():
if not self.__ray_metadata__.is_cross_language:
kwargs["_ray_trace_ctx"] = None
# Remove _ray_trace_ctx from kwargs if tracing disabled
kwargs.pop("_ray_trace_ctx", None)
return method(self, args, kwargs, *_args, **_kwargs)
class_name = self.__ray_metadata__.class_name