diff --git a/task_sdk/src/airflow/sdk/definitions/abstractoperator.py b/task_sdk/src/airflow/sdk/definitions/abstractoperator.py
index 5285bd97ef43..524ff78b727e 100644
--- 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
index ac50dcadbfc7..ee08bd19c908 100644
--- 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