diff --git a/hamilton/function_modifiers/macros.py b/hamilton/function_modifiers/macros.py index dfdb28b53..ae877db25 100644 --- a/hamilton/function_modifiers/macros.py +++ b/hamilton/function_modifiers/macros.py @@ -845,7 +845,7 @@ def final_result(p1: int, p2: int, p3: int) -> int: step(_add_two, y=source("upstream_node")).on_input("p2") ) def final_result(p1: int, p2: int, p3: int) -> int: - return upstream_int + return p1 + p2 + p3 We can also do this on the global level to set for all transforms a target parameter.