Thanks to visit codestin.com
Credit goes to github.com

Skip to content

Define TaskInstanceKey in task-sdk to support client server separation #60776

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
amoghrajesh merged 8 commits into from
Jan 24, 2026
Merged

Define TaskInstanceKey in task-sdk to support client server separation #60776

Changes from 1 commit
Commits
Show all changes
8 commits
Select commit Hold shift + click to select a range
464e3f9
Define TaskInstanceKey in task-sdk
amoghrajesh Jan 19, 2026
f2c3cde
handling compat in providers with compat sdk
amoghrajesh Jan 20, 2026
f9f6a61
handling compat in providers with compat sdk
amoghrajesh Jan 20, 2026
eb74240
handling compat in providers with compat sdk
amoghrajesh Jan 20, 2026
8b68c34
fixing mypy errors in providers
amoghrajesh Jan 21, 2026
dd9d5bf
fixing sdk tests
amoghrajesh Jan 21, 2026
4c3d94d
fixing dbx tests
amoghrajesh Jan 22, 2026
e4f6a06
fixing dbx tests
amoghrajesh Jan 22, 2026
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
Prev Previous commit
Next Next commit
fixing dbx tests
  • Loading branch information
@amoghrajesh
amoghrajesh committed Jan 22, 2026
commit 4c3d94db2799bc4494da840ae901f9fa17820637
9 changes: 5 additions & 4 deletions providers/databricks/src/airflow/providers/databricks/plugins/databricks_workflow.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -22,13 +22,14 @@

from airflow.exceptions import TaskInstanceNotFound
from airflow.models.dagrun import DagRun
from airflow.models.taskinstance import TaskInstance, TaskInstanceKey, clear_task_instances
from airflow.models.taskinstance import TaskInstance, clear_task_instances
from airflow.providers.common.compat.sdk import (
AirflowException,
AirflowOptionalProviderFeatureException,
AirflowPlugin,
BaseOperatorLink,
TaskGroup,
TaskInstanceKey,
XCom,
)
from airflow.providers.databricks.hooks.databricks import DatabricksHook
Expand Down Expand Up @@ -298,7 +299,7 @@ def xcom_key(self) -> str:
"""XCom key where the link is stored during task execution."""
return "databricks_job_run_link"

def get_link(
def get_link( # type: ignore[override] # Signature intentionally kept this way for Airflow 2.x compatibility
Comment thread
amoghrajesh marked this conversation as resolved.
self,
operator: BaseOperator,
dttm=None,
Expand Down Expand Up @@ -374,7 +375,7 @@ class WorkflowJobRepairAllFailedLink(BaseOperatorLink, LoggingMixin):

name = "Repair All Failed Tasks"

def get_link(
def get_link( # type: ignore[override] # Signature intentionally kept this way for Airflow 2.x compatibility
self,
operator,
dttm=None,
Expand Down Expand Up @@ -471,7 +472,7 @@ class WorkflowJobRepairSingleTaskLink(BaseOperatorLink, LoggingMixin):

name = "Repair a single task"

def get_link(
def get_link( # type: ignore[override] # Signature intentionally kept this way for Airflow 2.x compatibility
self,
operator,
dttm=None,
Expand Down
Loading