This commit is contained in:
William Fu-Hinthorn
2026-01-07 18:28:14 -08:00
parent 2d304b8a1b
commit eb9e6e2a0b
2 changed files with 2 additions and 5 deletions
+1 -5
View File
@@ -45,6 +45,7 @@ def _validate_traceable_config(raw: Any) -> TraceableConfig | None:
"process_inputs": raw.get("process_inputs"),
"process_outputs": raw.get("process_outputs"),
"enabled": raw.get("enabled"), # None means use external context
"wrapped": raw.get("wrapped"),
}
@@ -249,11 +250,6 @@ class PregelNode:
config = _validate_traceable_config(raw_config)
if config is None:
return None
unwrapped = getattr(func, "__wrapped__", None)
if unwrapped:
config["__unwrapped__"] = unwrapped
return config
@cached_property
@@ -50,6 +50,7 @@ def _set_traceable_config(
"process_inputs": process_inputs,
"process_outputs": process_outputs,
"enabled": enabled,
"wrapped": func,
},
)
return func