Fix import.

iza-institute-of-labor-economics · Nov 5, 2024 · efd88bd · efd88bd
1 parent 354507d
commit efd88bd
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/_gettsim/visualization.py b/src/_gettsim/visualization.py
@@ -11,8 +11,11 @@
 from pygments.formatters import HtmlFormatter
 
 from _gettsim.config import DEFAULT_TARGETS, TYPES_INPUT_VARIABLES
-from _gettsim.interface import check_functions_and_differentiate_types, set_up_dag
+from _gettsim.interface import set_up_dag
 from _gettsim.policy_environment import PolicyEnvironment
+from _gettsim.policy_environment_postprocessor import (
+    check_functions_and_differentiate_types,
+)
 from _gettsim.shared import (
     format_list_linewise,
     get_names_of_arguments_without_defaults,