Skip to content

Commit 73fa50d

Browse files
author
Jonathan Makunga
committed
Resolve PR comments
1 parent 80f0ca2 commit 73fa50d

File tree

1 file changed

+7
-1
lines changed

1 file changed

+7
-1
lines changed

src/sagemaker/serve/builder/jumpstart_builder.py

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -476,7 +476,13 @@ def _build_for_jumpstart(self):
476476
return self.pysdk_model
477477

478478
def _is_gated_model(self, model) -> bool:
479-
"""Determine if ``this`` Model is Gated"""
479+
"""Determine if ``this`` Model is Gated
480+
481+
Args:
482+
model (Model): Jumpstart Model
483+
Returns:
484+
bool: ``True`` if ``this`` Model is Gated
485+
"""
480486
s3_uri = model.model_data
481487
if isinstance(s3_uri, dict):
482488
s3_uri = s3_uri.get("S3DataSource").get("S3Uri")

0 commit comments

Comments
 (0)