From 29733c83dfbab87d46c936363fa6b6a37d961895 Mon Sep 17 00:00:00 2001
From: Benjamin Crabtree <bencrab@amazon.com>
Date: Mon, 18 Mar 2024 14:46:53 +0000
Subject: [PATCH] update
---
.../jumpstart/curated_hub/accessors/public_model_data.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
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: