You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@airflow.apache.org by hu...@apache.org on 2023/09/08 20:02:29 UTC
[airflow] branch main updated: Refactor: Consolidate import textwrap in providers (#34220)
This is an automated email from the ASF dual-hosted git repository.
husseinawala pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/airflow.git
The following commit(s) were added to refs/heads/main by this push:
new c5016f754d Refactor: Consolidate import textwrap in providers (#34220)
c5016f754d is described below
commit c5016f754df1b62046b9c1fce09574a69d8edebc
Author: Miroslav Šedivý <67...@users.noreply.github.com>
AuthorDate: Fri Sep 8 20:02:22 2023 +0000
Refactor: Consolidate import textwrap in providers (#34220)
---
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 5b8399ea13..a71905c97d 100644
--- a/airflow/providers/cncf/kubernetes/decorators/kubernetes.py
+++ b/airflow/providers/cncf/kubernetes/decorators/kubernetes.py
@@ -20,10 +20,10 @@ import base64
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 @@ class _KubernetesDecoratedOperator(DecoratedOperator, KubernetesPodOperator):
# 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 beb93fe113..30827bb6fe 100644
--- a/airflow/providers/docker/decorators/docker.py
+++ b/airflow/providers/docker/decorators/docker.py
@@ -20,8 +20,8 @@ import base64
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 @@ class _DockerDecoratedOperator(DecoratedOperator, DockerOperator):
# 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