fixup type error

astronomer · Nov 21, 2023 · 472287b · 472287b
1 parent 9ef16f9
commit 472287b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/cosmos/airflow/graph.py b/cosmos/airflow/graph.py
@@ -159,7 +159,7 @@ def generate_task_or_group(
     test_behavior: TestBehavior,
     test_indirect_selection: TestIndirectSelection,
     on_warning_callback: Callable[..., Any] | None,
-    node_config: dict[str, Any],
+    node_config: dict[str, Any] | None = None,
     **kwargs: Any,
 ) -> BaseOperator | TaskGroup | None:
     task_or_group: BaseOperator | TaskGroup | None = None

diff --git a/cosmos/core/airflow.py b/cosmos/core/airflow.py
@@ -12,7 +12,7 @@
 
 
 def get_airflow_task(
-    task: Task, dag: DAG, task_group: "TaskGroup | None" = None, extra_context: dict[str, Any] = {}
+    task: Task, dag: DAG, task_group: "TaskGroup | None" = None, extra_context: dict[str, Any] | None = None
 ) -> BaseOperator:
     """
     Get the Airflow Operator class for a Task.