sagemaker-python-sdk icon indicating copy to clipboard operation
sagemaker-python-sdk copied to clipboard

ModelBuilder cannot call `register()` method Without Setting `framework` and `framework_version` Attributes

Open plockxng opened this issue 1 month ago • 0 comments

Issue

Hello SageMaker Python SDK team,

It seems like based on this notebook, I can register a model directly without having to "build" it:

Image

What I'm observing is that it's missing a framework and framework_version attribute:

error 1: AttributeError: 'ModelBuilder' object has no attribute 'framework' error 2: AttributeError: 'ModelBuilder' object has no attribute 'framework_version'

This can be circumvented by setting these explicitly like so before calling the model_builder.register() method, but is there any way to remove this dependency, it seems unnecessary:

model_builder.framework = None
model_builder.framework_version = None

Resolution

Don't want to set these attributes explicitly, and most likely shouldn't be set by an end user.

plockxng avatar Dec 09 '25 19:12 plockxng