-
-
Notifications
You must be signed in to change notification settings - Fork 2k
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
Labels
Comments
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
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 theprocessor
argument of a Step in favor of passing the output ofprocessor.run
(estimator.fit
, ...) tostep_args
of the correspondingStep
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 theprocessor
/estimator
/...
.Happy make a PR adding this soon.
The text was updated successfully, but these errors were encountered: