Skip to content

Fix dynamic allocation specs handling for custom launcher - #38223

Merged
Taragolis merged 4 commits into
apache:mainfrom
sudohainguyen:cncf/unittest
Mar 18, 2024
Merged

Taragolis merged 4 commits into
apache:mainfrom
sudohainguyen:cncf/unittest

Conversation

@sudohainguyen

@sudohainguyen sudohainguyen commented Mar 17, 2024 •

Copy link
Copy Markdown
Contributor

Primarily about a fix to handle dynamicAllocation specs gracefully and adding unit test for custom object launcher.

Context:
When defining dynamic allocation specs, if users accidentally miss any required fields, currently we throw a KeyError exception which looks confused. So that I've put a minor change to make the exception thrown should be AirflowException with a more meaningful error message.

On the other hand, we also don't have unit test to cover some helper classes in custom_object_launcher.py, this PR also includes a few simple test cases.


^ 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.

@boring-cyborg boring-cyborg Bot added area:providers provider:cncf-kubernetes Kubernetes (k8s) provider related issues labels Mar 17, 2024
Comment thread tests/providers/cncf/kubernetes/operators/test_custom_object_launcher.py Outdated
Comment thread tests/providers/cncf/kubernetes/operators/test_custom_object_launcher.py Outdated
Comment thread tests/providers/cncf/kubernetes/operators/test_custom_object_launcher.py Outdated
Comment thread tests/providers/cncf/kubernetes/operators/test_custom_object_launcher.py Outdated
Comment thread airflow/providers/cncf/kubernetes/operators/custom_object_launcher.py Outdated
@sudohainguyen sudohainguyen changed the title Add unit test for Spark Kubernetes custom launcher Fix dynamic allocation specs handling for custom launcher Mar 17, 2024

@dirrao dirrao left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

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

LGTM

Signed-off-by: Hai Nguyen <quanghai.ng1512@gmail.com>
Signed-off-by: Hai Nguyen <quanghai.ng1512@gmail.com>
Signed-off-by: Hai Nguyen <quanghai.ng1512@gmail.com>
Signed-off-by: Hai Nguyen <quanghai.ng1512@gmail.com>
@Taragolis
Taragolis merged commit d4350a6 into apache:main Mar 18, 2024
@sudohainguyen
sudohainguyen deleted the cncf/unittest branch March 18, 2024 23:01
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

area:providers provider:cncf-kubernetes Kubernetes (k8s) provider related issues

Projects

None yet

Development

Successfully merging this pull request may close these issues.

5 participants