Skip to content

Commit

Permalink
Use correct argument for filter_overridden_functions.
Browse files Browse the repository at this point in the history
  • Loading branch information
MImmesberger committed Dec 5, 2024
1 parent 4568cf1 commit dfd4d91
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions src/_gettsim/policy_environment_postprocessor.py
Original file line number Diff line number Diff line change
Expand Up @@ -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.
Expand All @@ -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)
Expand Down

0 comments on commit dfd4d91

Please sign in to comment.