Skip to content

Commit

Permalink
fix: docker decorator pyi signature (apache#41509)
Browse files Browse the repository at this point in the history
  • Loading branch information
benbenbang authored Aug 15, 2024
1 parent 325f545 commit 393978d
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion airflow/decorators/__init__.pyi
Original file line number Diff line number Diff line change
Expand Up @@ -23,6 +23,7 @@ from __future__ import annotations
from datetime import timedelta
from typing import Any, Callable, Collection, Container, Iterable, Mapping, TypeVar, overload

from docker.types import Mount
from kubernetes.client import models as k8s

from airflow.decorators.base import FParams, FReturn, Task, TaskDecorator, _TaskDecorator
Expand Down Expand Up @@ -405,7 +406,7 @@ class TaskDecoratorCollection:
mount_tmp_dir: bool = True,
tmp_dir: str = "/tmp/airflow",
user: str | int | None = None,
mounts: list[str] | None = None,
mounts: list[Mount] | None = None,
entrypoint: str | list[str] | None = None,
working_dir: str | None = None,
xcom_all: bool = False,
Expand Down

0 comments on commit 393978d

Please sign in to comment.