Support tags in Model.register()

[athewsey]

Describe the feature you'd like

Model.register(...) should support adding tags, the same as Model.create(...) and pipelines RegisterModel already do.

How would this feature be used? Please describe.

I've been working with a customer who's registering models from a SageMaker Pipeline, using the PipelineSession syntax and ModelStep(step_args=model.register(...)).

We'd like to tag the package/version in Model Registry with the names of the training job and pipeline execution that generated it, since these seem not recorded automatically and are useful for maintaining lineage.

Describe alternatives you've considered

Without this parameter, it looks like we'll have to drop from ModelStep down to a RegisterModel step, which supports tagging... But the high-level .register() API is used in more samples and would be much preferred!

Additional context

As far as I can tell, the CreateModelPackage API already supports Tags - so I believe it should be no problem in the underlying APIs?

[julestalloen]

Are there any updates with regard to this feature request? Currently, the lack of a tags parameter forces us to use the RegisterModel step.