fix: Fix role parameter for _RepackModelStep

tuliocasagrande · tuliocasagrande · commit 7a1f4f818a38 · 2022-01-04T18:57:56.000-03:00
diff --git a/src/sagemaker/workflow/step_collections.py b/src/sagemaker/workflow/step_collections.py
@@ -336,7 +336,7 @@ def __init__(
                 depends_on=depends_on,
                 retry_policies=repack_model_step_retry_policies,
                 sagemaker_session=estimator.sagemaker_session,
-                role=estimator.sagemaker_session,
+                role=estimator.role,
                 model_data=model_data,
                 entry_point=entry_point,
                 source_dir=source_dir,
