Подсказка типа задачи, оформленной Airflow ⇐ Python

[Anonymous]

Возьмем этот простой пример, в котором одна задача принимает выходные данные другой:

Код: Выделить всё

import datetime
from airflow.decorators import dag, task

@task
def task_that_returns_a_string() -> str:
return "a string"

@task
def task_that_takes_a_string(arg: str) -> None:
print(arg)

@dag(
schedule_interval="@weekly",
start_date=datetime.datetime(2022, 1, 1),
)
def my_dag():
task_that_takes_a_string(task_that_returns_a_string())

my_dag()

Этот даг работает нормально, но мой редактор жалуется на строку Task_that_takes_a_string(task_that_returns_a_string()):

Код: Выделить всё

Argument of type "XComArg" cannot be assigned to parameter "arg" of type "str"
"XComArg" is incompatible with "str"PylancereportArgumentType

Конечно, я мог бы отключить проверку типов в этой строке, но тогда я бы упустил ценную обратную связь. Так что же мне следует изменить, чтобы получить правильную обратную связь по подсказкам типов?

Подробнее здесь: https://stackoverflow.com/questions/786 ... pe-hinting