This is an automated email from the ASF dual-hosted git repository.
zhongjiajie pushed a commit to branch main
in repository
https://gitbox.apache.org/repos/asf/dolphinscheduler-sdk-python.git
The following commit(s) were added to refs/heads/main by this push:
new 99d3ef5 CI: fix ruff lint and bulk ci error (#170)
99d3ef5 is described below
commit 99d3ef5299c584b74c89c8362f315772ca96199f
Author: Jay Chung <[email protected]>
AuthorDate: Tue Jan 20 14:39:00 2026 +0800
CI: fix ruff lint and bulk ci error (#170)
* fix error of ruff latest version
* tmp disable dead link check
* remove py 3.9 support
* use 3.10 build doc
---
.github/workflows/ci.yaml | 39 +++++++++-------------
setup.py | 1 +
src/pydolphinscheduler/configuration.py | 1 +
.../examples/task_dependent_example.py | 1 +
.../examples/task_sagemaker_example.py | 1 +
.../examples/task_sql_example.py | 1 +
.../examples/task_sub_workflow_example.py | 1 -
.../examples/tutorial_resource_plugin.py | 1 +
src/pydolphinscheduler/java_gateway.py | 1 +
src/pydolphinscheduler/models/datasource.py | 1 +
src/pydolphinscheduler/models/meta.py | 1 +
.../resources_plugin/__init__.py | 1 +
src/pydolphinscheduler/resources_plugin/github.py | 1 +
src/pydolphinscheduler/resources_plugin/gitlab.py | 1 +
src/pydolphinscheduler/resources_plugin/oss.py | 1 +
src/pydolphinscheduler/tasks/dependent.py | 1 +
src/pydolphinscheduler/tasks/dvc.py | 1 +
src/pydolphinscheduler/tasks/kubernetes.py | 1 +
src/pydolphinscheduler/tasks/mlflow.py | 1 +
src/pydolphinscheduler/tasks/pytorch.py | 1 +
src/pydolphinscheduler/tasks/sql.py | 1 +
src/pydolphinscheduler/tasks/switch.py | 1 +
src/pydolphinscheduler/utils/yaml_parser.py | 1 +
tests/core/test_engine.py | 1 -
tests/core/test_local_parameter.py | 1 -
tests/core/test_resource_definition.py | 1 +
tests/core/test_task.py | 1 +
tests/core/test_workflow.py | 1 +
tests/integration/test_java_gateway.py | 1 +
tests/integration/test_process_definition.py | 1 +
tests/integration/test_project.py | 1 +
tests/integration/test_tenant.py | 1 +
tests/models/test_database.py | 1 +
tests/resources_plugin/test_github.py | 1 +
tests/resources_plugin/test_gitlab.py | 1 +
tests/resources_plugin/test_local.py | 1 +
tests/resources_plugin/test_oss.py | 1 +
tests/resources_plugin/test_s3.py | 1 +
tests/tasks/test_condition.py | 1 +
tests/tasks/test_datax.py | 1 +
tests/tasks/test_dependent.py | 1 +
tests/tasks/test_dvc.py | 1 +
tests/tasks/test_mlflow.py | 1 +
tests/tasks/test_openmldb.py | 1 +
tests/tasks/test_python.py | 1 +
tests/tasks/test_pytorch.py | 1 +
tests/tasks/test_sagemaker.py | 1 +
tests/tasks/test_sql.py | 1 +
tests/tasks/test_sub_workflow.py | 1 +
tests/tasks/test_switch.py | 1 +
tests/testing/cli.py | 1 -
tests/testing/docker_wrapper.py | 1 +
tests/testing/file.py | 1 +
tests/testing/path.py | 1 +
tests/utils/test_date.py | 1 +
tests/utils/test_yaml_parser.py | 7 ++--
56 files changed, 69 insertions(+), 31 deletions(-)
diff --git a/.github/workflows/ci.yaml b/.github/workflows/ci.yaml
index e9ab2e5..37c5518 100644
--- a/.github/workflows/ci.yaml
+++ b/.github/workflows/ci.yaml
@@ -40,17 +40,17 @@ jobs:
runs-on: ubuntu-latest
steps:
- uses: apache/skywalking-eyes/header@main
- dead-link:
- runs-on: ubuntu-latest
- needs: license
- timeout-minutes: 30
- steps:
- - uses: actions/checkout@v3
- - run: sudo npm install -g [email protected]
- - run: |
- for file in $(find . -name "*.md"); do
- markdown-link-check -c .dlc.json -q "$file"
- done
+# dead-link:
+# runs-on: ubuntu-latest
+# needs: license
+# timeout-minutes: 30
+# steps:
+# - uses: actions/checkout@v3
+# - run: sudo npm install -g [email protected]
+# - run: |
+# for file in $(find . -name "*.md"); do
+# markdown-link-check -c .dlc.json -q "$file"
+# done
lint:
timeout-minutes: 15
runs-on: ubuntu-latest
@@ -76,7 +76,7 @@ jobs:
matrix:
# YAML parse `3.10` to `3.1`, so we have to add quotes for `'3.10'`,
see also:
#
https://github.com/actions/setup-python/issues/160#issuecomment-724485470
- python-version: ['3.9', '3.10', '3.11', '3.12', '3.13']
+ python-version: ['3.10', '3.11', '3.12', '3.13']
os: [ubuntu-latest, macOS-latest, windows-latest]
exclude:
# Skip because dependence [py4j](https://pypi.org/project/py4j/) not
work on those environments
@@ -86,13 +86,6 @@ jobs:
python-version: '3.11'
- os: windows-latest
python-version: '3.12'
- # Python 3.9 is on macos-13 but not macos-latest (macos-14-arm64)
- #
https://github.com/actions/setup-python/issues/696#issuecomment-1637587760
- - os: macos-latest
- python-version: '3.9'
- include:
- - python-version: '3.9'
- os: macos-13
steps:
- uses: actions/checkout@v3
- name: Set up Python ${{ matrix.python-version }}
@@ -133,7 +126,7 @@ jobs:
- name: Set up Python
uses: actions/setup-python@v4
with:
- python-version: 3.11
+ python-version: '3.10'
- name: Install Dependences
run: |
python -m pip install --upgrade ${{ env.DEPENDENCES }}
@@ -218,15 +211,15 @@ jobs:
name: CI
if: always()
needs:
- - dead-link
+# - dead-link
- local-ci
- integrate-test
runs-on: ubuntu-latest
steps:
- name: Status
+# if [[ ${{ needs.dead-link.result }} != 'success' ]] || \
run: |
- if [[ ${{ needs.dead-link.result }} != 'success' ]] || \
- [[ ${{ needs.local-ci.result }} != 'success' ]] || \
+ if [[ ${{ needs.local-ci.result }} != 'success' ]] || \
([[ ${{ needs.integrate-test.result }} != 'skipped' ]] && [[ ${{
needs.integrate-test.result }} != 'success' ]]); then
echo "CI Failed!"
exit -1
diff --git a/setup.py b/setup.py
index dcc83db..d687567 100644
--- a/setup.py
+++ b/setup.py
@@ -16,6 +16,7 @@
# under the License.
"""The script for setting up pydolphinscheduler."""
+
from __future__ import annotations
import logging
diff --git a/src/pydolphinscheduler/configuration.py
b/src/pydolphinscheduler/configuration.py
index 2a73446..e7a2842 100644
--- a/src/pydolphinscheduler/configuration.py
+++ b/src/pydolphinscheduler/configuration.py
@@ -16,6 +16,7 @@
# under the License.
"""Configuration module for pydolphinscheduler."""
+
import logging
import os
from pathlib import Path
diff --git a/src/pydolphinscheduler/examples/task_dependent_example.py
b/src/pydolphinscheduler/examples/task_dependent_example.py
index a09aaf1..bf43ca5 100644
--- a/src/pydolphinscheduler/examples/task_dependent_example.py
+++ b/src/pydolphinscheduler/examples/task_dependent_example.py
@@ -35,6 +35,7 @@ task_dependent:
task_dependent(this task dependent on task_dependent_external.task_1 and
task_dependent_external.task_2).
"""
+
from pydolphinscheduler import configuration
from pydolphinscheduler.core.workflow import Workflow
from pydolphinscheduler.tasks.dependent import And, Dependent, DependentItem,
Or
diff --git a/src/pydolphinscheduler/examples/task_sagemaker_example.py
b/src/pydolphinscheduler/examples/task_sagemaker_example.py
index 6024823..1ddff49 100644
--- a/src/pydolphinscheduler/examples/task_sagemaker_example.py
+++ b/src/pydolphinscheduler/examples/task_sagemaker_example.py
@@ -17,6 +17,7 @@
# [start workflow_declare]
"""A example workflow for task sagemaker."""
+
import json
from pydolphinscheduler.core.workflow import Workflow
diff --git a/src/pydolphinscheduler/examples/task_sql_example.py
b/src/pydolphinscheduler/examples/task_sql_example.py
index f2a8bd4..ca1319d 100644
--- a/src/pydolphinscheduler/examples/task_sql_example.py
+++ b/src/pydolphinscheduler/examples/task_sql_example.py
@@ -18,6 +18,7 @@
# [start workflow_declare]
"""A example workflow for task SQL."""
+
from pathlib import Path
from pydolphinscheduler.core.workflow import Workflow
diff --git a/src/pydolphinscheduler/examples/task_sub_workflow_example.py
b/src/pydolphinscheduler/examples/task_sub_workflow_example.py
index df7b6ce..c7417fb 100644
--- a/src/pydolphinscheduler/examples/task_sub_workflow_example.py
+++ b/src/pydolphinscheduler/examples/task_sub_workflow_example.py
@@ -26,7 +26,6 @@ from pydolphinscheduler.core.workflow import Workflow
from pydolphinscheduler.tasks.sub_workflow import SubWorkflow
from pydolphinscheduler.tasks.shell import Shell
-
# [start workflow_declare]
# [start sub_workflow_declare]
with Workflow(name="sub_workflow_downstream") as wf_downstream, Workflow(
diff --git a/src/pydolphinscheduler/examples/tutorial_resource_plugin.py
b/src/pydolphinscheduler/examples/tutorial_resource_plugin.py
index 0089e5f..1b56d5d 100644
--- a/src/pydolphinscheduler/examples/tutorial_resource_plugin.py
+++ b/src/pydolphinscheduler/examples/tutorial_resource_plugin.py
@@ -22,6 +22,7 @@ Resource plug-ins can be defined in workflows and tasks
it will instantiate and run all the task it have.
"""
+
import os
from pathlib import Path
diff --git a/src/pydolphinscheduler/java_gateway.py
b/src/pydolphinscheduler/java_gateway.py
index e848475..5230066 100644
--- a/src/pydolphinscheduler/java_gateway.py
+++ b/src/pydolphinscheduler/java_gateway.py
@@ -16,6 +16,7 @@
# under the License.
"""Module java gateway, contain gateway behavior."""
+
from __future__ import annotations
import contextlib
diff --git a/src/pydolphinscheduler/models/datasource.py
b/src/pydolphinscheduler/models/datasource.py
index a7407a2..390b3d3 100644
--- a/src/pydolphinscheduler/models/datasource.py
+++ b/src/pydolphinscheduler/models/datasource.py
@@ -16,6 +16,7 @@
# under the License.
"""Module database."""
+
from __future__ import annotations
import json
diff --git a/src/pydolphinscheduler/models/meta.py
b/src/pydolphinscheduler/models/meta.py
index f4da69c..0aa4c77 100644
--- a/src/pydolphinscheduler/models/meta.py
+++ b/src/pydolphinscheduler/models/meta.py
@@ -22,6 +22,7 @@ This module contains the ModelMeta class, which is used to
convert ``py4j.java_g
server to get some resource from database, but you want to make sure the
return object is a in Python
object.
"""
+
from __future__ import annotations
from functools import wraps
diff --git a/src/pydolphinscheduler/resources_plugin/__init__.py
b/src/pydolphinscheduler/resources_plugin/__init__.py
index 1e24e1e..5195d89 100644
--- a/src/pydolphinscheduler/resources_plugin/__init__.py
+++ b/src/pydolphinscheduler/resources_plugin/__init__.py
@@ -16,6 +16,7 @@
# under the License.
"""Init resources_plugin package."""
+
from pydolphinscheduler.resources_plugin.github import GitHub
from pydolphinscheduler.resources_plugin.gitlab import GitLab
from pydolphinscheduler.resources_plugin.local import Local
diff --git a/src/pydolphinscheduler/resources_plugin/github.py
b/src/pydolphinscheduler/resources_plugin/github.py
index f103d75..825d35e 100644
--- a/src/pydolphinscheduler/resources_plugin/github.py
+++ b/src/pydolphinscheduler/resources_plugin/github.py
@@ -16,6 +16,7 @@
# under the License.
"""DolphinScheduler github resource plugin."""
+
from __future__ import annotations
import base64
diff --git a/src/pydolphinscheduler/resources_plugin/gitlab.py
b/src/pydolphinscheduler/resources_plugin/gitlab.py
index d1ee80b..1c1ddc1 100644
--- a/src/pydolphinscheduler/resources_plugin/gitlab.py
+++ b/src/pydolphinscheduler/resources_plugin/gitlab.py
@@ -16,6 +16,7 @@
# under the License.
"""DolphinScheduler gitlab resource plugin."""
+
from __future__ import annotations
from urllib.parse import urljoin, urlparse
diff --git a/src/pydolphinscheduler/resources_plugin/oss.py
b/src/pydolphinscheduler/resources_plugin/oss.py
index 871b8db..819b5d6 100644
--- a/src/pydolphinscheduler/resources_plugin/oss.py
+++ b/src/pydolphinscheduler/resources_plugin/oss.py
@@ -16,6 +16,7 @@
# under the License.
"""DolphinScheduler oss resource plugin."""
+
from __future__ import annotations
from urllib.parse import urljoin, urlparse
diff --git a/src/pydolphinscheduler/tasks/dependent.py
b/src/pydolphinscheduler/tasks/dependent.py
index 2894373..5bf1dbb 100644
--- a/src/pydolphinscheduler/tasks/dependent.py
+++ b/src/pydolphinscheduler/tasks/dependent.py
@@ -16,6 +16,7 @@
# under the License.
"""Task dependent."""
+
from __future__ import annotations
import warnings
diff --git a/src/pydolphinscheduler/tasks/dvc.py
b/src/pydolphinscheduler/tasks/dvc.py
index 46b2125..a2e9464 100644
--- a/src/pydolphinscheduler/tasks/dvc.py
+++ b/src/pydolphinscheduler/tasks/dvc.py
@@ -16,6 +16,7 @@
# under the License.
"""Task dvc."""
+
from __future__ import annotations
from copy import deepcopy
diff --git a/src/pydolphinscheduler/tasks/kubernetes.py
b/src/pydolphinscheduler/tasks/kubernetes.py
index 0152cb2..21133ce 100644
--- a/src/pydolphinscheduler/tasks/kubernetes.py
+++ b/src/pydolphinscheduler/tasks/kubernetes.py
@@ -16,6 +16,7 @@
# under the License.
"""Task Kubernetes."""
+
from pydolphinscheduler.constants import TaskType
from pydolphinscheduler.core.task import BatchTask
diff --git a/src/pydolphinscheduler/tasks/mlflow.py
b/src/pydolphinscheduler/tasks/mlflow.py
index 2d03709..d9cf0fc 100644
--- a/src/pydolphinscheduler/tasks/mlflow.py
+++ b/src/pydolphinscheduler/tasks/mlflow.py
@@ -16,6 +16,7 @@
# under the License.
"""Task mlflow."""
+
from __future__ import annotations
from copy import deepcopy
diff --git a/src/pydolphinscheduler/tasks/pytorch.py
b/src/pydolphinscheduler/tasks/pytorch.py
index db0d9be..1ce622f 100644
--- a/src/pydolphinscheduler/tasks/pytorch.py
+++ b/src/pydolphinscheduler/tasks/pytorch.py
@@ -16,6 +16,7 @@
# under the License.
"""Task Pytorch."""
+
from __future__ import annotations
from pydolphinscheduler.constants import TaskType
diff --git a/src/pydolphinscheduler/tasks/sql.py
b/src/pydolphinscheduler/tasks/sql.py
index c9c3f49..1a038c3 100644
--- a/src/pydolphinscheduler/tasks/sql.py
+++ b/src/pydolphinscheduler/tasks/sql.py
@@ -16,6 +16,7 @@
# under the License.
"""Task sql."""
+
from __future__ import annotations
import logging
diff --git a/src/pydolphinscheduler/tasks/switch.py
b/src/pydolphinscheduler/tasks/switch.py
index 625ac70..61fd5a5 100644
--- a/src/pydolphinscheduler/tasks/switch.py
+++ b/src/pydolphinscheduler/tasks/switch.py
@@ -16,6 +16,7 @@
# under the License.
"""Task Switch."""
+
from __future__ import annotations
from pydolphinscheduler.constants import TaskType
diff --git a/src/pydolphinscheduler/utils/yaml_parser.py
b/src/pydolphinscheduler/utils/yaml_parser.py
index a79dfec..ea36f0e 100644
--- a/src/pydolphinscheduler/utils/yaml_parser.py
+++ b/src/pydolphinscheduler/utils/yaml_parser.py
@@ -16,6 +16,7 @@
# under the License.
"""YAML parser utils, parser yaml string to ``ruamel.yaml`` object and nested
key dict."""
+
from __future__ import annotations
import copy
diff --git a/tests/core/test_engine.py b/tests/core/test_engine.py
index 312f27a..5d0a918 100644
--- a/tests/core/test_engine.py
+++ b/tests/core/test_engine.py
@@ -17,7 +17,6 @@
"""Test Task Engine."""
-
from unittest.mock import patch
import pytest
diff --git a/tests/core/test_local_parameter.py
b/tests/core/test_local_parameter.py
index f1042d2..b0e17c0 100644
--- a/tests/core/test_local_parameter.py
+++ b/tests/core/test_local_parameter.py
@@ -17,7 +17,6 @@
"""Test parameter."""
-
import pytest
from pydolphinscheduler.core.parameter import (
diff --git a/tests/core/test_resource_definition.py
b/tests/core/test_resource_definition.py
index 07fcac3..97539cf 100644
--- a/tests/core/test_resource_definition.py
+++ b/tests/core/test_resource_definition.py
@@ -16,6 +16,7 @@
# under the License.
"""Test resource definition."""
+
import pytest
from pydolphinscheduler.core.resource import Resource
diff --git a/tests/core/test_task.py b/tests/core/test_task.py
index 8b7d2a0..6f14267 100644
--- a/tests/core/test_task.py
+++ b/tests/core/test_task.py
@@ -16,6 +16,7 @@
# under the License.
"""Test Task class function."""
+
from __future__ import annotations
import logging
diff --git a/tests/core/test_workflow.py b/tests/core/test_workflow.py
index 66748ed..15d0b9c 100644
--- a/tests/core/test_workflow.py
+++ b/tests/core/test_workflow.py
@@ -16,6 +16,7 @@
# under the License.
"""Test workflow."""
+
from __future__ import annotations
import warnings
diff --git a/tests/integration/test_java_gateway.py
b/tests/integration/test_java_gateway.py
index 4545546..1b33eb0 100644
--- a/tests/integration/test_java_gateway.py
+++ b/tests/integration/test_java_gateway.py
@@ -16,6 +16,7 @@
# under the License.
"""Test pydolphinscheduler java gateway."""
+
import importlib
import warnings
from unittest.mock import PropertyMock, patch
diff --git a/tests/integration/test_process_definition.py
b/tests/integration/test_process_definition.py
index 42aa96a..babf9cc 100644
--- a/tests/integration/test_process_definition.py
+++ b/tests/integration/test_process_definition.py
@@ -16,6 +16,7 @@
# under the License.
"""Test workflow in integration."""
+
from __future__ import annotations
import pytest
diff --git a/tests/integration/test_project.py
b/tests/integration/test_project.py
index 7b8d875..b8e1105 100644
--- a/tests/integration/test_project.py
+++ b/tests/integration/test_project.py
@@ -16,6 +16,7 @@
# under the License.
"""Test pydolphinscheduler project."""
+
import pytest
from pydolphinscheduler.models import Project, User
diff --git a/tests/integration/test_tenant.py b/tests/integration/test_tenant.py
index 844f8d7..1432fa3 100644
--- a/tests/integration/test_tenant.py
+++ b/tests/integration/test_tenant.py
@@ -16,6 +16,7 @@
# under the License.
"""Test pydolphinscheduler tenant."""
+
import pytest
from pydolphinscheduler.models import Tenant, User
diff --git a/tests/models/test_database.py b/tests/models/test_database.py
index 44e7561..3e063d2 100644
--- a/tests/models/test_database.py
+++ b/tests/models/test_database.py
@@ -16,6 +16,7 @@
# under the License.
"""Test Database."""
+
import json
from unittest.mock import patch
diff --git a/tests/resources_plugin/test_github.py
b/tests/resources_plugin/test_github.py
index 1f1a631..bc6f6f9 100644
--- a/tests/resources_plugin/test_github.py
+++ b/tests/resources_plugin/test_github.py
@@ -16,6 +16,7 @@
# under the License.
"""Test github resource plugin."""
+
from unittest.mock import PropertyMock, patch
import pytest
diff --git a/tests/resources_plugin/test_gitlab.py
b/tests/resources_plugin/test_gitlab.py
index 6bb90ac..d884647 100644
--- a/tests/resources_plugin/test_gitlab.py
+++ b/tests/resources_plugin/test_gitlab.py
@@ -16,6 +16,7 @@
# under the License.
"""Test github resource plugin."""
+
import pytest
from pydolphinscheduler.resources_plugin.gitlab import GitLab
diff --git a/tests/resources_plugin/test_local.py
b/tests/resources_plugin/test_local.py
index 82b196f..0fca83d 100644
--- a/tests/resources_plugin/test_local.py
+++ b/tests/resources_plugin/test_local.py
@@ -16,6 +16,7 @@
# under the License.
"""Test local resource plugin."""
+
from pathlib import Path
from unittest.mock import PropertyMock, patch
diff --git a/tests/resources_plugin/test_oss.py
b/tests/resources_plugin/test_oss.py
index 7e57e82..bc53fb3 100644
--- a/tests/resources_plugin/test_oss.py
+++ b/tests/resources_plugin/test_oss.py
@@ -16,6 +16,7 @@
# under the License.
"""Test oss resource plugin."""
+
import pytest
from pydolphinscheduler.resources_plugin.oss import OSS
diff --git a/tests/resources_plugin/test_s3.py
b/tests/resources_plugin/test_s3.py
index 5f75f3e..1880aa2 100644
--- a/tests/resources_plugin/test_s3.py
+++ b/tests/resources_plugin/test_s3.py
@@ -16,6 +16,7 @@
# under the License.
"""Test oss resource plugin."""
+
import pytest
from pydolphinscheduler.resources_plugin import S3
diff --git a/tests/tasks/test_condition.py b/tests/tasks/test_condition.py
index 41dbdb2..3e85646 100644
--- a/tests/tasks/test_condition.py
+++ b/tests/tasks/test_condition.py
@@ -16,6 +16,7 @@
# under the License.
"""Test Task dependent."""
+
from __future__ import annotations
from unittest.mock import patch
diff --git a/tests/tasks/test_datax.py b/tests/tasks/test_datax.py
index aa884a6..85835f1 100644
--- a/tests/tasks/test_datax.py
+++ b/tests/tasks/test_datax.py
@@ -16,6 +16,7 @@
# under the License.
"""Test Task DataX."""
+
from pathlib import Path
from unittest.mock import patch
diff --git a/tests/tasks/test_dependent.py b/tests/tasks/test_dependent.py
index 0e41a9c..fb10ac5 100644
--- a/tests/tasks/test_dependent.py
+++ b/tests/tasks/test_dependent.py
@@ -16,6 +16,7 @@
# under the License.
"""Test Task dependent."""
+
from __future__ import annotations
import itertools
diff --git a/tests/tasks/test_dvc.py b/tests/tasks/test_dvc.py
index a297228..357512a 100644
--- a/tests/tasks/test_dvc.py
+++ b/tests/tasks/test_dvc.py
@@ -16,6 +16,7 @@
# under the License.
"""Test Task Dvc."""
+
from unittest.mock import patch
from pydolphinscheduler.tasks.dvc import DVCDownload, DVCInit, DvcTaskType,
DVCUpload
diff --git a/tests/tasks/test_mlflow.py b/tests/tasks/test_mlflow.py
index 9e6517e..ba0a693 100644
--- a/tests/tasks/test_mlflow.py
+++ b/tests/tasks/test_mlflow.py
@@ -16,6 +16,7 @@
# under the License.
"""Test Task MLflow."""
+
from copy import deepcopy
from unittest.mock import patch
diff --git a/tests/tasks/test_openmldb.py b/tests/tasks/test_openmldb.py
index 5675bce..51c9087 100644
--- a/tests/tasks/test_openmldb.py
+++ b/tests/tasks/test_openmldb.py
@@ -16,6 +16,7 @@
# under the License.
"""Test Task OpenMLDB."""
+
from unittest.mock import patch
from pydolphinscheduler.tasks.openmldb import OpenMLDB
diff --git a/tests/tasks/test_python.py b/tests/tasks/test_python.py
index e842f90..2f6afc9 100644
--- a/tests/tasks/test_python.py
+++ b/tests/tasks/test_python.py
@@ -16,6 +16,7 @@
# under the License.
"""Test Task python."""
+
from pathlib import Path
from unittest.mock import patch
diff --git a/tests/tasks/test_pytorch.py b/tests/tasks/test_pytorch.py
index 66c1664..55d53ff 100644
--- a/tests/tasks/test_pytorch.py
+++ b/tests/tasks/test_pytorch.py
@@ -16,6 +16,7 @@
# under the License.
"""Test Task Pytorch."""
+
from copy import deepcopy
from unittest.mock import patch
diff --git a/tests/tasks/test_sagemaker.py b/tests/tasks/test_sagemaker.py
index ec0c42e..e29f862 100644
--- a/tests/tasks/test_sagemaker.py
+++ b/tests/tasks/test_sagemaker.py
@@ -16,6 +16,7 @@
# under the License.
"""Test Task SageMaker."""
+
import json
from unittest.mock import patch
diff --git a/tests/tasks/test_sql.py b/tests/tasks/test_sql.py
index f150820..3ff8128 100644
--- a/tests/tasks/test_sql.py
+++ b/tests/tasks/test_sql.py
@@ -16,6 +16,7 @@
# under the License.
"""Test Task Sql."""
+
from pathlib import Path
from unittest.mock import patch
diff --git a/tests/tasks/test_sub_workflow.py b/tests/tasks/test_sub_workflow.py
index 9c4d454..2ea522a 100644
--- a/tests/tasks/test_sub_workflow.py
+++ b/tests/tasks/test_sub_workflow.py
@@ -16,6 +16,7 @@
# under the License.
"""Test Task sub workflow."""
+
import warnings
from unittest.mock import patch
diff --git a/tests/tasks/test_switch.py b/tests/tasks/test_switch.py
index e9a24bf..b6e8537 100644
--- a/tests/tasks/test_switch.py
+++ b/tests/tasks/test_switch.py
@@ -16,6 +16,7 @@
# under the License.
"""Test Task switch."""
+
from __future__ import annotations
from unittest.mock import patch
diff --git a/tests/testing/cli.py b/tests/testing/cli.py
index 0d2c1d1..089c1d0 100644
--- a/tests/testing/cli.py
+++ b/tests/testing/cli.py
@@ -17,7 +17,6 @@
"""Utils of command line test."""
-
from click.testing import CliRunner
from tests.testing.constants import DEV_MODE
diff --git a/tests/testing/docker_wrapper.py b/tests/testing/docker_wrapper.py
index ce8ba0c..d7bade3 100644
--- a/tests/testing/docker_wrapper.py
+++ b/tests/testing/docker_wrapper.py
@@ -16,6 +16,7 @@
# under the License.
"""Wrap docker commands for easier create docker container."""
+
from __future__ import annotations
import logging
diff --git a/tests/testing/file.py b/tests/testing/file.py
index 67fd4ba..86da010 100644
--- a/tests/testing/file.py
+++ b/tests/testing/file.py
@@ -16,6 +16,7 @@
# under the License.
"""Testing util about file operating."""
+
from __future__ import annotations
from pathlib import Path
diff --git a/tests/testing/path.py b/tests/testing/path.py
index 0359a23..413bef9 100644
--- a/tests/testing/path.py
+++ b/tests/testing/path.py
@@ -16,6 +16,7 @@
# under the License.
"""Handle path related issue in test module."""
+
from __future__ import annotations
from collections.abc import Generator
diff --git a/tests/utils/test_date.py b/tests/utils/test_date.py
index 9599d61..03a7a20 100644
--- a/tests/utils/test_date.py
+++ b/tests/utils/test_date.py
@@ -16,6 +16,7 @@
# under the License.
"""Test utils.date module."""
+
from __future__ import annotations
from datetime import datetime, timedelta
diff --git a/tests/utils/test_yaml_parser.py b/tests/utils/test_yaml_parser.py
index c017067..a0f7d18 100644
--- a/tests/utils/test_yaml_parser.py
+++ b/tests/utils/test_yaml_parser.py
@@ -16,6 +16,7 @@
# under the License.
"""Test utils.path_dict module."""
+
from __future__ import annotations
import pytest
@@ -67,8 +68,7 @@ expects = [
},
]
-param = [
- """#example
+param = ["""#example
name:
# details
family: Smith # very common
@@ -76,8 +76,7 @@ name:
mark:
name_mark:
key: value
-"""
-]
+"""]
with open(path_default_config_yaml) as f:
param.append(f.read())
