Skip to content

Commit 9346011

Browse files
committed
chore: new class/field names
1 parent 1a6e184 commit 9346011

File tree

3 files changed

+6
-6
lines changed

3 files changed

+6
-6
lines changed

src/sagemaker/jumpstart/estimator.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -35,7 +35,7 @@
3535
from sagemaker.jumpstart.factory.estimator import get_deploy_kwargs, get_fit_kwargs, get_init_kwargs
3636
from sagemaker.jumpstart.factory.model import get_default_predictor
3737
from sagemaker.jumpstart.session_utils import get_model_id_version_from_training_job
38-
from sagemaker.jumpstart.types import JumpStartInternalMetadata
38+
from sagemaker.jumpstart.types import JumpStartModelInternalConfig
3939
from sagemaker.jumpstart.utils import (
4040
validate_model_id_and_get_type,
4141
resolve_model_sagemaker_config_field,
@@ -535,7 +535,7 @@ def _validate_model_id_and_get_type_hook():
535535
if not self.model_type and not hub_arn:
536536
raise ValueError(INVALID_MODEL_ID_ERROR_MSG.format(model_id=model_id))
537537

538-
self._internal = JumpStartInternalMetadata(
538+
self._internal_config = JumpStartModelInternalConfig(
539539
specs=verify_model_region_and_return_specs(
540540
region=self.region,
541541
model_id=self.model_id,

src/sagemaker/jumpstart/model.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,7 @@
3838
get_register_kwargs,
3939
)
4040
from sagemaker.jumpstart.session_utils import get_model_id_version_from_endpoint
41-
from sagemaker.jumpstart.types import JumpStartInternalMetadata, JumpStartSerializablePayload
41+
from sagemaker.jumpstart.types import JumpStartModelInternalConfig, JumpStartSerializablePayload
4242
from sagemaker.jumpstart.utils import (
4343
validate_model_id_and_get_type,
4444
verify_model_region_and_return_specs,
@@ -312,7 +312,7 @@ def _validate_model_id_and_type():
312312
if not self.model_type and not hub_arn:
313313
raise ValueError(INVALID_MODEL_ID_ERROR_MSG.format(model_id=model_id))
314314

315-
self._internal = JumpStartInternalMetadata(
315+
self._internal_config = JumpStartModelInternalConfig(
316316
specs=verify_model_region_and_return_specs(
317317
region=self.region,
318318
model_id=self.model_id,

src/sagemaker/jumpstart/types.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2484,13 +2484,13 @@ def __init__(
24842484
self.accept_eula = accept_eula
24852485

24862486

2487-
class JumpStartInternalMetadata(JumpStartDataHolderType):
2487+
class JumpStartModelInternalConfig(JumpStartDataHolderType):
24882488
"""Data class for storing internal/private fields for JumpStart models."""
24892489

24902490
slots = ["specs"]
24912491

24922492
def __init__(self, specs: JumpStartModelSpecs):
2493-
"""Initializes a JumpStartInternalMetadata object.
2493+
"""Initializes a JumpStartModelInternalConfig object.
24942494
24952495
Args:
24962496
specs (JumpStartModelSpecs): specs for model.

0 commit comments

Comments
 (0)