PySDK Version
Describe the bug
A clear and concise description of what the bug is.
ModelStep expects the Model class to have a sagemaker_session attribute but the spec does not allow it
To reproduce
Provide a Model instance as the model argument of the register model step arguments
Expected behavior
ModelStep can be instantiated with a repack step
Screenshots or logs
If applicable, add screenshots or logs to help explain your problem.
System information
A description of your system. Please provide:
- SageMaker Python SDK version: 3
- Framework name (eg. PyTorch) or algorithm (eg. KMeans): SKLearn
- Framework version: 1.4.2
- Python version: 3.11
- CPU or GPU: CPU
- Custom Docker image (Y/N): N
Additional context
Add any other context about the problem here.
PySDK Version
Describe the bug
A clear and concise description of what the bug is.
ModelStep expects the Model class to have a
sagemaker_sessionattribute but the spec does not allow itTo reproduce
Provide a Model instance as the
modelargument of the register model step argumentsExpected behavior
ModelStep can be instantiated with a repack step
Screenshots or logs
If applicable, add screenshots or logs to help explain your problem.
System information
A description of your system. Please provide:
Additional context
Add any other context about the problem here.