Skip to content
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

Fix overriding of loader_args task resolver in papermill plugin #2660

Merged
merged 3 commits into from
Aug 8, 2024

Conversation

eapolinario
Copy link
Collaborator

@eapolinario eapolinario commented Aug 7, 2024

Why are the changes needed?

In #1859 (which was released in flytekitplugins-papermill 1.10.0) we ended up overriding the task resolver's loader_args method in order to generate the correct NotebookTask container arguments. However we never restored that function to its original value, which has the side effect that in the case of shared task resolvers (e.g. the default_task_resolver used for regular tasks) we end up reusing the last NotebookTask registered as the task name.

What changes were proposed in this pull request?

This PR restores the loader_args in papermill plugin's get_container and get_k8s_pod.

How was this patch tested?

Setup process

Screenshots

Check all the applicable boxes

  • I updated the documentation accordingly.
  • All new and existing tests passed.
  • All commits are signed-off.

Related PRs

Docs link

Signed-off-by: Eduardo Apolinario <[email protected]>
Signed-off-by: Eduardo Apolinario <[email protected]>
@eapolinario eapolinario changed the title Add repro test case for papermill issue Fix overriding of loader_args task resolver in papermill plugin Aug 7, 2024
Copy link
Member

@pingsutw pingsutw left a comment

Choose a reason for hiding this comment

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

lgtm, thank you!

@eapolinario eapolinario merged commit 6e04c11 into master Aug 8, 2024
99 checks passed
mao3267 pushed a commit to mao3267/flytekit that referenced this pull request Aug 9, 2024
…eorg#2660)

* Add repro test case

Signed-off-by: Eduardo Apolinario <[email protected]>

* Restore loader_args in papermill plugin

Signed-off-by: Eduardo Apolinario <[email protected]>

* Add unit tests

Signed-off-by: Eduardo Apolinario <[email protected]>

---------

Signed-off-by: Eduardo Apolinario <[email protected]>
Co-authored-by: Eduardo Apolinario <[email protected]>
Signed-off-by: mao3267 <[email protected]>
Mecoli1219 pushed a commit to Mecoli1219/flytekit that referenced this pull request Aug 14, 2024
…eorg#2660)

* Add repro test case

Signed-off-by: Eduardo Apolinario <[email protected]>

* Restore loader_args in papermill plugin

Signed-off-by: Eduardo Apolinario <[email protected]>

* Add unit tests

Signed-off-by: Eduardo Apolinario <[email protected]>

---------

Signed-off-by: Eduardo Apolinario <[email protected]>
Co-authored-by: Eduardo Apolinario <[email protected]>
Mecoli1219 pushed a commit to Mecoli1219/flytekit that referenced this pull request Aug 14, 2024
…eorg#2660)

* Add repro test case

Signed-off-by: Eduardo Apolinario <[email protected]>

* Restore loader_args in papermill plugin

Signed-off-by: Eduardo Apolinario <[email protected]>

* Add unit tests

Signed-off-by: Eduardo Apolinario <[email protected]>

---------

Signed-off-by: Eduardo Apolinario <[email protected]>
Co-authored-by: Eduardo Apolinario <[email protected]>
Signed-off-by: Mecoli1219 <[email protected]>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants