[dagster-airlift] [api-docs] mapping functions

dagster-io · Nov 6, 2024 · 65537ea · 65537ea
1 parent 74ed6d5
commit 65537ea
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 3 deletions.
diff --git a/docs/sphinx/sections/api/apidocs/libraries/dagster-airlift.rst b/docs/sphinx/sections/api/apidocs/libraries/dagster-airlift.rst
@@ -17,28 +17,37 @@ AirflowInstance
 
 .. autoclass:: AirflowBasicAuthBackend 
 
-
 Assets & Definitions
 ^^^^^^^^^^^^^^^^^^^^
 
 .. autofunction:: build_defs_from_airflow_instance
 
+Mapping Dagster assets to Airflow tasks/dags:
+~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
+
+.. autofunction:: assets_with_task_mappings
+
+.. autofunction:: assets_with_dag_mappings
+
+.. autofunction:: targeted_by_multiple_tasks
+
 Annotations for customizable components:
 ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
 
 .. autoclass:: DagSelectorFn
 
 .. autoclass:: DagsterEventTransformerFn
 
+.. autoclass:: TaskHandleDict
+
 Objects for retrieving information about the Airflow/Dagster mapping:
 ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
 
 .. autoclass:: DagInfo
 
 .. autoclass:: AirflowDefinitionsData
 
-
-
+.. currentmodule:: dagster_airlift.mwaa
 
 MWAA (dagster_airlift.mwaa)
 ---------------------------
@@ -47,3 +56,5 @@ MWAA (dagster_airlift.mwaa)
 .. autoclass:: MwaaSessionAuthBackend
 
 
+
+
diff --git a/examples/experimental/dagster-airlift/dagster_airlift/core/__init__.py b/examples/experimental/dagster-airlift/dagster_airlift/core/__init__.py
@@ -8,6 +8,10 @@
     build_airflow_mapped_defs as build_airflow_mapped_defs,
     build_defs_from_airflow_instance as build_defs_from_airflow_instance,
 )
+from .multiple_tasks import (
+    TaskHandleDict as TaskHandleDict,
+    targeted_by_multiple_tasks as targeted_by_multiple_tasks,
+)
 from .sensor.event_translation import (
     AssetEvent as AssetEvent,
     DagsterEventTransformerFn as DagsterEventTransformerFn,