Python QueryRunner.recreate_udf Examples

Example #1

File: transformation_dag_builder.py Project: teamdatatonic/dop

def exec_recreate_udf(task, runner: QueryRunner, **kwargs):
    airflow_context = kwargs["airflow_context"]
    relation = create_relation_from_task(task=task)
    sql = airflow_context["templates_dict"]["sql"]

    arguments = task.options.get("arguments", [])
    runner.recreate_udf(arguments=arguments, query=sql, relation=relation)

Example #2

def exec_recreate_udf(task, runner: QueryRunner, **kwargs):
    airflow_context = kwargs['airflow_context']
    relation = create_relation_from_task(task=task)
    sql = airflow_context['templates_dict']['sql']

    arguments = task.options.get('arguments', [])
    runner.recreate_udf(
        arguments=arguments,
        query=sql,
        relation=relation
    )