Fix Mypy errors in main after recent changes (apache#43920)

boschglobal · Nov 12, 2024 · 3b14684 · 3b14684
1 parent f71df97
commit 3b14684
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/providers/tests/microsoft/conftest.py b/providers/tests/microsoft/conftest.py
@@ -137,6 +137,7 @@ def xcom_pull(
             key: str = XCOM_RETURN_KEY,
             include_prior_dates: bool = False,
             session: Session = NEW_SESSION,
+            run_id: str | None = None,
             *,
             map_indexes: Iterable[int] | int | None = None,
             default: Any | None = None,

diff --git a/providers/tests/system/snowflake/example_snowpark_decorator.py b/providers/tests/system/snowflake/example_snowpark_decorator.py
@@ -65,7 +65,7 @@ def setup_data(session: Session):
         df.write.save_as_table(table_name, mode="overwrite")
         return table_name
 
-    table_name = setup_data()  # type: ignore[call-arg]
+    table_name = setup_data()  # type: ignore[call-arg, misc]
 
     @task.snowpark
     def check_num_rows(table_name: str):