repackModel could expose an output_path parameter

[tuliocasagrande]

Describe the feature you'd like
Right now, repackModel steps write to root of the default bucket. Here's one line from test code that we can verify that:

sagemaker-python-sdk/tests/unit/sagemaker/workflow/test_utils.py

Line 143 in f2fd421

"OutputDataConfig": {"S3OutputPath": f"s3://{BUCKET}/"},

It would be great if we could expose an output_path and perhaps also a code_location parameters to help us organize our S3 buckets.

How would this feature be used? Please describe.
The feature would be used to help us organize S3 buckets into namespaces/hierachy. For example, right now I organize my prefixes as:

s3://<myprojectbucket>/<base_job_prefix>/preprocess/...
s3://<myprojectbucket>/<base_job_prefix>/train/...
s3://<myprojectbucket>/<base_job_prefix>/evaluate/...

But since repackModel doesn't expose parameters to tune S3 outputs, I also end up some objects in the root, example:

s3://<myprojectbucket>/pipelines-<jobid-1>-sklearnRepackModel.../output/model.tar.gz
s3://<myprojectbucket>/pipelines-<jobid-2>-sklearnRepackModel.../output/model.tar.gz
s3://<myprojectbucket>/pipelines-<jobid-N>-sklearnRepackModel.../output/model.tar.gz
s3://<myprojectbucket>/sagemaker-scikit-learn-<timestamp-1>/source/sourcedir.tar.gz
s3://<myprojectbucket>/sagemaker-scikit-learn-<timestamp-2>/source/sourcedir.tar.gz
s3://<myprojectbucket>/sagemaker-scikit-learn-<timestamp-N>/source/sourcedir.tar.gz

Describe alternatives you've considered
I'd need some guidance to find all the correct places, but I can submit a PR with these changes:

1. step_collections.RegisterModel and step_collections.EstimatorTransformer, add a new parameter called repack_output_path
2. On _RepackModelStep() calls (L139, L183, L324), handover the parameter repack_output_path=repack_output_path.
3. Inside of _RepackModelStep, on the SKLearn() call, add output_path=repack_output_path and code_location=repack_output_path.
4. Maybe another place I didn't identify.

I'd also need help to understand which unit tests would be needed, on RegisterModel, EstimatorTransformer, _RepackModelStep, or all of them.

Thanks for your review!

[jerrypeng7773]

@tuliocasagrande looks like your already unblocked yourself =)

but for future reference, RegisterModel takes kwargs, and you will be able to pass outpout_path as a kwarg to it, and the repack model training job will use it as its OutputDataConfig.S3Outputpath.

Closing this issue, feel free to re-open if you have any other concern.