From 55bb907ae6110817feb74841de51e9bb722526e2 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Miroslav=20=C5=A0ediv=C3=BD?=
 <6774676+eumiro@users.noreply.github.com>
Date: Fri, 8 Sep 2023 20:34:44 +0200
Subject: [PATCH] Refactor: Consolidate import textwrap in providers

---
 airflow/providers/cncf/kubernetes/decorators/kubernetes.py | 4 ++--
 airflow/providers/docker/decorators/docker.py              | 4 ++--
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/airflow/providers/cncf/kubernetes/decorators/kubernetes.py b/airflow/providers/cncf/kubernetes/decorators/kubernetes.py
index 5b8399ea13770..a71905c97d2dd 100644
--- a/airflow/providers/cncf/kubernetes/decorators/kubernetes.py
+++ b/airflow/providers/cncf/kubernetes/decorators/kubernetes.py
@@ -20,10 +20,10 @@
 import inspect
 import os
 import pickle
+import textwrap
 import uuid
 from shlex import quote
 from tempfile import TemporaryDirectory
-from textwrap import dedent
 from typing import TYPE_CHECKING, Callable, Sequence
 
 import dill
@@ -80,7 +80,7 @@ def __init__(self, namespace: str = "default", use_dill: bool = False, **kwargs)
     # TODO: Remove me once this provider min supported Airflow version is 2.6
     def get_python_source(self):
         raw_source = inspect.getsource(self.python_callable)
-        res = dedent(raw_source)
+        res = textwrap.dedent(raw_source)
         res = remove_task_decorator(res, self.custom_operator_name)
         return res
 
diff --git a/airflow/providers/docker/decorators/docker.py b/airflow/providers/docker/decorators/docker.py
index beb93fe1139de..30827bb6fefd6 100644
--- a/airflow/providers/docker/decorators/docker.py
+++ b/airflow/providers/docker/decorators/docker.py
@@ -20,8 +20,8 @@
 import inspect
 import os
 import pickle
+import textwrap
 from tempfile import TemporaryDirectory
-from textwrap import dedent
 from typing import TYPE_CHECKING, Callable, Sequence
 
 import dill
@@ -139,7 +139,7 @@ def execute(self, context: Context):
     # TODO: Remove me once this provider min supported Airflow version is 2.6
     def get_python_source(self):
         raw_source = inspect.getsource(self.python_callable)
-        res = dedent(raw_source)
+        res = textwrap.dedent(raw_source)
         res = remove_task_decorator(res, self.custom_operator_name)
         return res