Publish pytask.tree_util. (#426)

Author: tobiasraabe

src/pytask/tree_util.py

@@ -0,0 +1,19 @@
+"""Publishes functions of :mod:`_pytask.tree_util`."""
+from __future__ import annotations
+
+from _pytask.tree_util import PyTree
+from _pytask.tree_util import tree_flatten_with_path
+from _pytask.tree_util import tree_leaves
+from _pytask.tree_util import tree_map
+from _pytask.tree_util import tree_map_with_path
+from _pytask.tree_util import tree_structure
+
+
+__all__ = [
+    "tree_flatten_with_path",
+    "tree_leaves",
+    "tree_map",
+    "tree_map_with_path",
+    "tree_structure",
+    "PyTree",
+]