diff --git a/src/sagemaker/jumpstart/curated_hub/accessors/public_model_data.py b/src/sagemaker/jumpstart/curated_hub/accessors/public_model_data.py index 9894ad05e0..a4e339591b 100644 --- a/src/sagemaker/jumpstart/curated_hub/accessors/public_model_data.py +++ b/src/sagemaker/jumpstart/curated_hub/accessors/public_model_data.py @@ -86,7 +86,7 @@ def default_training_dataset_s3_reference(self) -> S3ObjectLocation: """Retrieves s3 reference for s3 directory containing model training datasets""" if not self.model_specs.training_supported: return None - return S3ObjectLocation(self._get_bucket_name(), self.__get_training_dataset_prefix()) + return S3ObjectLocation(self._get_bucket_name(), self._get_training_dataset_prefix()) @property def demo_notebook_s3_reference(self) -> S3ObjectLocation: