Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 1 addition & 0 deletions docs/spelling_wordlist.txt
Original file line number Diff line number Diff line change
Expand Up @@ -911,6 +911,7 @@ js
Json
json
jsonl
JsonValue
juli
Jupyter
jupyter
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -31,7 +31,7 @@
from airflow.providers.common.compat.sdk import BaseOperator, conf

try:
from airflow.sdk.bases.resumablemixin import ResumableJobMixin
from airflow.sdk import ResumableJobMixin
except ImportError:
# Airflow 2 compat.
# ResumableJobMixin does not exist in Airflow 2, so we need to add a stub to make it
Expand Down
4 changes: 2 additions & 2 deletions task-sdk/src/airflow/sdk/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -126,7 +126,7 @@
cross_downstream,
)
from airflow.sdk.bases.operatorlink import BaseOperatorLink
from airflow.sdk.bases.resumablemixin import ResumableJobMixin
from airflow.sdk.bases.resumablejobmixin import ResumableJobMixin
from airflow.sdk.bases.sensor import BaseSensorOperator, PokeReturnValue
from airflow.sdk.bases.skipmixin import SkipMixin
from airflow.sdk.bases.xcom import BaseXCom
Expand Down Expand Up @@ -256,7 +256,7 @@
"PokeReturnValue": ".bases.sensor",
"ProductMapper": ".definitions.partition_mappers.product",
"QuarterWindow": ".definitions.partition_mappers.window",
"ResumableJobMixin": ".bases.resumablemixin",
"ResumableJobMixin": ".bases.resumablejobmixin",
"RetryAction": ".definitions.retry_policy",
"RetryDecision": ".definitions.retry_policy",
"RetryPolicy": ".definitions.retry_policy",
Expand Down
2 changes: 2 additions & 0 deletions task-sdk/src/airflow/sdk/__init__.pyi
Original file line number Diff line number Diff line change
Expand Up @@ -35,6 +35,7 @@ from airflow.sdk.bases.operator import (
cross_downstream as cross_downstream,
)
from airflow.sdk.bases.operatorlink import BaseOperatorLink as BaseOperatorLink
from airflow.sdk.bases.resumablejobmixin import ResumableJobMixin as ResumableJobMixin
from airflow.sdk.bases.sensor import (
BaseSensorOperator as BaseSensorOperator,
PokeReturnValue as PokeReturnValue,
Expand Down Expand Up @@ -169,6 +170,7 @@ __all__ = [
"RetryDecision",
"RetryPolicy",
"RetryRule",
"ResumableJobMixin",
"RollupMapper",
"SecretCache",
"SkipMixin",
Expand Down
2 changes: 1 addition & 1 deletion task-sdk/tests/task_sdk/bases/test_resumablemixin.py
Original file line number Diff line number Diff line change
Expand Up @@ -20,8 +20,8 @@

import pytest

from airflow.sdk import ResumableJobMixin
from airflow.sdk.bases.operator import BaseOperator
from airflow.sdk.bases.resumablemixin import ResumableJobMixin

if TYPE_CHECKING:
from pydantic import JsonValue
Expand Down
Loading