From 393978d65c968b0f07af8cb75cd60151e5043b6e Mon Sep 17 00:00:00 2001
From: Ben Chen <bn@bitbrew.dev>
Date: Fri, 16 Aug 2024 00:38:35 +0200
Subject: [PATCH] fix: docker decorator pyi signature (#41509)

---
 airflow/decorators/__init__.pyi | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/airflow/decorators/__init__.pyi b/airflow/decorators/__init__.pyi
index e6d247918395..7dd887431c6f 100644
--- a/airflow/decorators/__init__.pyi
+++ b/airflow/decorators/__init__.pyi
@@ -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
@@ -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,