Skip to content

Commit b2d281c

Browse files
committed
fix: reset cache if content bucket changes
1 parent 7e96f0e commit b2d281c

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

src/sagemaker/jumpstart/utils.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -100,6 +100,7 @@ def get_jumpstart_content_bucket(
100100
accessors.JumpStartModelsAccessor.set_jumpstart_content_bucket(bucket_to_return)
101101

102102
if bucket_to_return != old_content_bucket:
103+
accessors.JumpStartModelsAccessor.reset_cache()
103104
for info_log in info_logs:
104105
constants.JUMPSTART_LOGGER.info(info_log)
105106
return bucket_to_return

0 commit comments

Comments
 (0)