Task SDK: Fix typing for autoregistered_dags

Fixing one of the TODOs: apache#43076 This was already addressed.
astronomer · Oct 30, 2024 · 47accc3 · 47accc3
1 parent b94b1a1
commit 47accc3
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/task_sdk/src/airflow/sdk/definitions/abstractoperator.py b/task_sdk/src/airflow/sdk/definitions/abstractoperator.py
@@ -87,7 +87,7 @@ class AbstractOperator(DAGNode):
 
     outlets: list
     inlets: list
-    # TODO:
+
     trigger_rule: TriggerRule
     _needs_expansion: bool | None = None
     _on_failure_fail_dagrun = False

diff --git a/task_sdk/src/airflow/sdk/definitions/contextmanager.py b/task_sdk/src/airflow/sdk/definitions/contextmanager.py
@@ -94,8 +94,7 @@ class DagContext(ContextStack[DAG]):
 
     """
 
-    # TODO: Task-SDK, should module type be optional? Will that break more?
-    autoregistered_dags: set[tuple[DAG, ModuleType | None]] = set()
+    autoregistered_dags: set[tuple[DAG, ModuleType]] = set()
     current_autoregister_module_name: str | None = None
 
     @classmethod