|
33 | 33 | from sagemaker.jumpstart.hub.utils import (
|
34 | 34 | get_hub_model_version,
|
35 | 35 | get_info_from_hub_resource_arn,
|
| 36 | + create_hub_bucket_if_it_does_not_exist, |
| 37 | + generate_default_hub_bucket_name, |
36 | 38 | create_s3_object_reference_from_uri,
|
37 | 39 | construct_hub_arn_from_name,
|
38 | 40 | )
|
@@ -88,11 +90,12 @@ def _fetch_hub_bucket_name(self) -> str:
|
88 | 90 | if hub_output_location:
|
89 | 91 | location = create_s3_object_reference_from_uri(hub_output_location)
|
90 | 92 | return location.bucket
|
| 93 | + default_bucket_name = generate_default_hub_bucket_name(self._sagemaker_session) |
91 | 94 | JUMPSTART_LOGGER.warning(
|
92 |
| - "There is not a Hub bucket associated with %s.", |
| 95 | + "There is not a Hub bucket associated with %s. Using %s", |
93 | 96 | self.hub_name,
|
94 | 97 | )
|
95 |
| - return None |
| 98 | + return default_bucket_name |
96 | 99 | except exceptions.ClientError:
|
97 | 100 | JUMPSTART_LOGGER.warning(
|
98 | 101 | "There is not a Hub bucket associated with %s.",
|
@@ -127,18 +130,16 @@ def create(
|
127 | 130 | ) -> Dict[str, str]:
|
128 | 131 | """Creates a hub with the given description"""
|
129 | 132 |
|
130 |
| - s3_storage_config = ( |
131 |
| - {"S3OutputPath": self.hub_storage_location.get_uri()} |
132 |
| - if self.hub_storage_location |
133 |
| - else None |
| 133 | + create_hub_bucket_if_it_does_not_exist( |
| 134 | + self.hub_storage_location.bucket, self._sagemaker_session |
134 | 135 | )
|
135 | 136 |
|
136 | 137 | return self._sagemaker_session.create_hub(
|
137 | 138 | hub_name=self.hub_name,
|
138 | 139 | hub_description=description,
|
139 | 140 | hub_display_name=display_name,
|
140 | 141 | hub_search_keywords=search_keywords,
|
141 |
| - s3_storage_config=s3_storage_config, |
| 142 | + s3_storage_config={"S3OutputPath": self.hub_storage_location.get_uri()}, |
142 | 143 | tags=tags,
|
143 | 144 | )
|
144 | 145 |
|
|
0 commit comments