From dfd4d91f21b87d3b135776af75a9be88898510f2 Mon Sep 17 00:00:00 2001 From: Marvin Immesberger Date: Thu, 5 Dec 2024 14:09:13 +0100 Subject: [PATCH] Use correct argument for filter_overridden_functions. --- src/_gettsim/policy_environment_postprocessor.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/_gettsim/policy_environment_postprocessor.py b/src/_gettsim/policy_environment_postprocessor.py index 7fbb25d7a..cc49b05f4 100644 --- a/src/_gettsim/policy_environment_postprocessor.py +++ b/src/_gettsim/policy_environment_postprocessor.py @@ -107,13 +107,13 @@ def add_derived_functions_to_functions_tree( def filter_overridden_functions( - functions: dict[str, Any], data: dict[str, Any] + all_functions: dict[str, Any], data: dict[str, Any] ) -> tuple[dict[str, Any], dict[str, Any]]: """Filter functions that are overridden by input columns. Parameters ---------- - functions : dict + all_functions : dict Dictionary containing functions to build the DAG. data : dict Dictionary containing the input columns. @@ -130,7 +130,7 @@ def filter_overridden_functions( functions_overridden = {} names_of_columns_in_data = tree_flatten_with_qualified_name(data)[0] - function_paths, functions_leafs, _ = tree_flatten_with_path(functions) + function_paths, functions_leafs, _ = tree_flatten_with_path(all_functions) for name, func in zip(function_paths, functions_leafs): qualified_name = "__".join(name)