Skip to content

Nuking extra code branching in task runner for inlets, outlets#46302

Merged
amoghrajesh merged 3 commits intoapache:mainfrom
astronomer:inlet-outlet-check-improvement
Jan 31, 2025
Merged

Nuking extra code branching in task runner for inlets, outlets#46302
amoghrajesh merged 3 commits intoapache:mainfrom
astronomer:inlet-outlet-check-improvement

Conversation

@amoghrajesh
Copy link
Contributor

The logic to check for task inlets, outlets in task runner had unwanted code branching - if, else when it could have been simpler. (Yea, i wrote the initial code, but didnt realise that time :)).

This PR simplifies that part by keeping only the if and adding a finally. Additionally, renamed msg that received OKResponse as ok_response to not confuse with the actual message from supervsior.

Extended the tests for OK = true as well as false.


^ Add meaningful description above
Read the Pull Request Guidelines for more information.
In case of fundamental code changes, an Airflow Improvement Proposal (AIP) is needed.
In case of a new dependency, check compliance with the ASF 3rd Party License Policy.
In case of backwards incompatible changes please leave a note in a newsfragment file, named {pr_number}.significant.rst or {issue_number}.significant.rst, in newsfragments.

outlet_events=outlet_events,
)
ok_response = SUPERVISOR_COMMS.get_message() # type: ignore
if isinstance(ok_response, OKResponse) and not ok_response.ok:
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Suggested change
if isinstance(ok_response, OKResponse) and not ok_response.ok:
if not isinstance(ok_response, OKResponse) or not ok_response.ok:

Just in case we get some other object type back?

amoghrajesh and others added 2 commits January 31, 2025 15:00
Co-authored-by: Ash Berlin-Taylor <ash_github@firemirror.com>
@amoghrajesh amoghrajesh merged commit 357ef74 into apache:main Jan 31, 2025
62 checks passed
@amoghrajesh amoghrajesh deleted the inlet-outlet-check-improvement branch January 31, 2025 12:11
Prab-27 pushed a commit to Prab-27/airflow that referenced this pull request Jan 31, 2025
…e#46302)

Co-authored-by: Ash Berlin-Taylor <ash_github@firemirror.com>
niklasr22 pushed a commit to niklasr22/airflow that referenced this pull request Feb 8, 2025
…e#46302)

Co-authored-by: Ash Berlin-Taylor <ash_github@firemirror.com>
ambika-garg pushed a commit to ambika-garg/airflow that referenced this pull request Feb 17, 2025
…e#46302)

Co-authored-by: Ash Berlin-Taylor <ash_github@firemirror.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants