diff --git a/src/sagemaker/jumpstart/estimator.py b/src/sagemaker/jumpstart/estimator.py index eae91f2a8d..7595288512 100644 --- a/src/sagemaker/jumpstart/estimator.py +++ b/src/sagemaker/jumpstart/estimator.py @@ -705,7 +705,7 @@ def attach( training job. """ - return cls._attach( + return Estimator._attach( training_job_name=training_job_name, sagemaker_session=sagemaker_session, model_channel_name=model_channel_name,