airflow.providers.snowflake.utils.snowpark

Module Contents

Functions

inject_session_into_op_kwargs(python_callable, ...)

Inject Snowpark session into operator kwargs based on signature of python callable.

airflow.providers.snowflake.utils.snowpark.inject_session_into_op_kwargs(python_callable, op_kwargs, session)[source]

Inject Snowpark session into operator kwargs based on signature of python callable.

If there is a keyword argument named session in the signature of the python callable, a Snowpark session object will be injected into kwargs.

Parameters
  • python_callable (Callable) – Python callable

  • op_kwargs (dict) – Operator kwargs

  • session (snowflake.snowpark.Session | None) – Snowpark session

Was this entry helpful?