From b550d3b8ae3a92da86ad0f31ad574dd8e3a4718f Mon Sep 17 00:00:00 2001 From: Chris DeCarolis Date: Fri, 1 Nov 2024 11:39:57 -0700 Subject: [PATCH] [dagster-airlift] [api-docs] mapping functions --- .../api/apidocs/libraries/dagster-airlift.rst | 13 +++++++++---- .../dagster_airlift/core/__init__.py | 4 ++++ 2 files changed, 13 insertions(+), 4 deletions(-) diff --git a/docs/sphinx/sections/api/apidocs/libraries/dagster-airlift.rst b/docs/sphinx/sections/api/apidocs/libraries/dagster-airlift.rst index b386b9fe87934..0035d8e2d3079 100644 --- a/docs/sphinx/sections/api/apidocs/libraries/dagster-airlift.rst +++ b/docs/sphinx/sections/api/apidocs/libraries/dagster-airlift.rst @@ -17,23 +17,26 @@ AirflowInstance .. autoclass:: BasicAirflowAuthBackend -.. currentmodule:: dagster_airlift.mwaa - 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) --------------------------- @@ -41,3 +44,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 index b4eacbb84556c..03cb4f7d3fbb9 100644 --- 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,