diff --git a/providers/tests/microsoft/conftest.py b/providers/tests/microsoft/conftest.py index de25d24fb05e1..6ac4c0a03a368 100644 --- 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 index 59cf1f1fedbb6..1e4c0d4c1f02a 100644 --- 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):