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

Add SageMaker PipelineSession mock #7037

Open
stiebels opened this issue Nov 17, 2023 · 1 comment
Open

Add SageMaker PipelineSession mock #7037

stiebels opened this issue Nov 17, 2023 · 1 comment

Comments

@stiebels
Copy link
Contributor

stiebels commented Nov 17, 2023

Hi @bblommers

SageMaker Pipelines has been updating their interface between Steps and Processors/Estimators/Tuners/... . They're now deprecating the old way of passing an estimator/processor/tuner object to the processor argument of a Step in favor of passing the output of processor.run (estimator.fit, ...) to step_args of the corresponding Step class (e.g. see here or here).

Currently, this is not supported in moto and trying to mock a Step that uses this interface results in an error (can c/p stack trace here later if needed; would outline in the PR eventually anyway).

In order to start supporting this, we'll need to add a mock for PipelineSession as an instance thereof is required to be passed to the processor/estimator/....

Happy make a PR adding this soon.

@bblommers
Copy link
Collaborator

Thanks for raising this @stiebels - a PR would be very welcome. Let me know if you need any help!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

No branches or pull requests

2 participants