File tree Expand file tree Collapse file tree 2 files changed +2
-2
lines changed
src/sagemaker/jumpstart/curated_hub
tests/unit/sagemaker/jumpstart/curated_hub Expand file tree Collapse file tree 2 files changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -307,7 +307,7 @@ def get_jumpstart_model_and_version(
307
307
]
308
308
return JumpStartModelInfo (model_id = jumpstart_model_id , version = jumpstart_model_version )
309
309
310
-
310
+
311
311
def is_gated_bucket (bucket_name : str ) -> bool :
312
312
"""Returns true if the bucket name is the JumpStart gated bucket."""
313
313
return bucket_name in constants .JUMPSTART_GATED_BUCKET_NAME_SET
Original file line number Diff line number Diff line change @@ -181,7 +181,7 @@ def test_is_gated_bucket():
181
181
assert utils .is_gated_bucket ("jumpstart-private-cache-prod-us-east-1" ) is True
182
182
183
183
assert utils .is_gated_bucket ("jumpstart-cache-prod-us-west-2" ) is False
184
-
184
+
185
185
assert utils .is_gated_bucket ("" ) is False
186
186
187
187
You can’t perform that action at this time.
0 commit comments