File tree Expand file tree Collapse file tree 3 files changed +6
-6
lines changed Expand file tree Collapse file tree 3 files changed +6
-6
lines changed Original file line number Diff line number Diff line change 35
35
from sagemaker .jumpstart .factory .estimator import get_deploy_kwargs , get_fit_kwargs , get_init_kwargs
36
36
from sagemaker .jumpstart .factory .model import get_default_predictor
37
37
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
39
39
from sagemaker .jumpstart .utils import (
40
40
validate_model_id_and_get_type ,
41
41
resolve_model_sagemaker_config_field ,
@@ -535,7 +535,7 @@ def _validate_model_id_and_get_type_hook():
535
535
if not self .model_type and not hub_arn :
536
536
raise ValueError (INVALID_MODEL_ID_ERROR_MSG .format (model_id = model_id ))
537
537
538
- self ._internal = JumpStartInternalMetadata (
538
+ self ._internal_config = JumpStartModelInternalConfig (
539
539
specs = verify_model_region_and_return_specs (
540
540
region = self .region ,
541
541
model_id = self .model_id ,
Original file line number Diff line number Diff line change 38
38
get_register_kwargs ,
39
39
)
40
40
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
42
42
from sagemaker .jumpstart .utils import (
43
43
validate_model_id_and_get_type ,
44
44
verify_model_region_and_return_specs ,
@@ -312,7 +312,7 @@ def _validate_model_id_and_type():
312
312
if not self .model_type and not hub_arn :
313
313
raise ValueError (INVALID_MODEL_ID_ERROR_MSG .format (model_id = model_id ))
314
314
315
- self ._internal = JumpStartInternalMetadata (
315
+ self ._internal_config = JumpStartModelInternalConfig (
316
316
specs = verify_model_region_and_return_specs (
317
317
region = self .region ,
318
318
model_id = self .model_id ,
Original file line number Diff line number Diff line change @@ -2484,13 +2484,13 @@ def __init__(
2484
2484
self .accept_eula = accept_eula
2485
2485
2486
2486
2487
- class JumpStartInternalMetadata (JumpStartDataHolderType ):
2487
+ class JumpStartModelInternalConfig (JumpStartDataHolderType ):
2488
2488
"""Data class for storing internal/private fields for JumpStart models."""
2489
2489
2490
2490
slots = ["specs" ]
2491
2491
2492
2492
def __init__ (self , specs : JumpStartModelSpecs ):
2493
- """Initializes a JumpStartInternalMetadata object.
2493
+ """Initializes a JumpStartModelInternalConfig object.
2494
2494
2495
2495
Args:
2496
2496
specs (JumpStartModelSpecs): specs for model.
You can’t perform that action at this time.
0 commit comments