Namespaces

.gitignore

@@ -84,6 +84,9 @@ ipython_config.py
 # VS Code
 .vscode
 
+# IntelliJ
+.idea/
+
 # pyenv
 .python-version
 

environment.yml

@@ -30,3 +30,7 @@ dependencies:
   # Development
   - jupyterlab
   - nbsphinx
+
+  # Pip
+  - pip:
+    - flatten-dict

src/dags/__init__.py

@@ -1,8 +1,12 @@
 from dags.dag import concatenate_functions
 from dags.dag import get_ancestors
+from dags.dag_tree import concatenate_functions_tree
+from dags.dag_tree import create_input_structure_tree
 
 
 __all__ = [
     "concatenate_functions",
+    "concatenate_functions_tree",
+    "create_input_structure_tree",
     "get_ancestors",
 ]

src/dags/dag.py

@@ -319,7 +319,7 @@ def _create_arguments_of_concatenated_function(functions, dag):
         dag (networkx.DiGraph): The complete DAG.
 
     Returns:
-        inspect.Signature: The signature of the concatenated function.
+        list: The arguments of the concatenated function.
 
     """
     function_names = set(functions)