Skip to content

Update Sagemaker Operators and Sensors to inherit Base AWS classes#50321

Merged
eladkal merged 4 commits into
apache:mainfrom
ellisms:base-sagemaker
May 12, 2025
Merged

Update Sagemaker Operators and Sensors to inherit Base AWS classes#50321
eladkal merged 4 commits into
apache:mainfrom
ellisms:base-sagemaker

Conversation

@ellisms

@ellisms ellisms commented May 7, 2025

Copy link
Copy Markdown
Contributor

Updated the operators and sensors. Did not update triggers, as those are a little more complex.

Partially addresses #35278

@eladkal

eladkal commented May 7, 2025

Copy link
Copy Markdown
Contributor

We also have trigger class for sagemaker don't we?

@ellisms

ellisms commented May 8, 2025

Copy link
Copy Markdown
Contributor Author

There is, but the changes will be a bit more involved. I wasn't sure when I could get to those, so I went ahead and published the Operators/Sensors since they are done.

@eladkal

eladkal commented May 8, 2025

Copy link
Copy Markdown
Contributor

There is, but the changes will be a bit more involved. I wasn't sure when I could get to those, so I went ahead and published the Operators/Sensors since they are done.

No problem.
Can you fix the failing tests?

@ellisms

ellisms commented May 8, 2025

Copy link
Copy Markdown
Contributor Author

Yeah, odd those are failing. Will double check.

@eladkal eladkal merged commit 24d041d into apache:main May 12, 2025
sanederchik pushed a commit to sanederchik/airflow that referenced this pull request Jun 7, 2025
…pache#50321)

* Update SageMaker operators to inherit AwsBaseOperator

* Update SageMaker sensors to inherit AwsBaseSensor

* Update docs

* Fixed sensor tests
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