-
Notifications
You must be signed in to change notification settings - Fork 14.7k
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.
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
AIP-72: Task SDK support for on_task_instance_* listeners, make OpenLineage compatible #45294
Conversation
f07a78e
to
81b886b
Compare
@mobuchowski - can you please rebase that one -> we found and issue with @jscheffl with the new caching scheme - fixed in #45347 that would run "main" version of the tests. I am asking in all affected PRs to rebase. |
81b886b
to
15fd95a
Compare
Actually - I rebased it now. |
I will rebase very soon as I'm working on some of the test failures anyway 🙂 |
35a94e2
to
86297af
Compare
86297af
to
6d88389
Compare
c2e2165
to
305fff0
Compare
3dc5ae4
to
39172c3
Compare
@mobuchowski You had also mentioned that you would be raising another PR once #45732 was merged. Since that is already merged, is that follow-up PR already in progress. Thank you! |
39172c3
to
0705a85
Compare
… TaskSDK, make OpenLineage provider support Airflow 3's listener interface Signed-off-by: Maciej Obuchowski <[email protected]>
0705a85
to
61da140
Compare
Hey @vikramkoka - rebased this PR and fixed conflicts, it should be good to merge. Haven't started on followup to #45732 but will try to find some time next week. I don't think we need to hold up this PR for that, especially since rebasing 2000 lines PR on fast moving target takes significant amount of time. |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
I think this works. Should have a follow-up PR changing OL identifier for DagRun and TaskInstance to use run_id.
So far cleaning airflow installation only happened in canary runs and it caused some PRs not failing when they should - for exmaple the apache#45294 was green when it should fail because uuid6 package was not removed before installing old version of Airlfow. Cleaning airflow installation is fast with uv so we should be ok with running it always for compatibility tests.
So far cleaning airflow installation only happened in canary runs and it caused some PRs not failing when they should - for exmaple the apache#45294 was green when it should fail because uuid6 package was not removed before installing old version of Airlfow. Cleaning airflow installation is fast with uv so we should be ok with running it always for compatibility tests.
…46693) So far cleaning airflow installation only happened in canary runs and it caused some PRs not failing when they should - for exmaple the #45294 was green when it should fail because uuid6 package was not removed before installing old version of Airlfow. Cleaning airflow installation is fast with uv so we should be ok with running it always for compatibility tests.
… TaskSDK, make OpenLineage provider support Airflow 3's listener interface (apache#45294) Signed-off-by: Maciej Obuchowski <[email protected]>
…pache#46693) So far cleaning airflow installation only happened in canary runs and it caused some PRs not failing when they should - for exmaple the apache#45294 was green when it should fail because uuid6 package was not removed before installing old version of Airlfow. Cleaning airflow installation is fast with uv so we should be ok with running it always for compatibility tests.
With AIP-72, there is no access to the database session from the worker process, and the runtime objects have some differences to the db models. This PR contains three commits that deal with that situation:
on_task_instance_*
listeners interface to AIP-72: dropssession
argument and makestask_instance
argument an instance ofRuntimeTaskInstance
class, not database modelSome followup work:
Activity
to make logging better visible from UI, and distinct from task logsRuntimeTaskInstance
)closes #45423