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

[plockxng]

Issue

Hello SageMaker Python SDK team,

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

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.

[mohamedzeidan2021]

Fixed. ModelBuilder now defaults framework and framework_version to None at construction (in __post_init__), so register() can be called directly without building first — no more AttributeError, and the manual model_builder.framework = None workaround is no longer needed. Landed via #5558 and present in the current SDK. Closing as resolved.