fix: _internal_config definition

Author: evakravi

src/sagemaker/jumpstart/estimator.py

@@ -535,18 +535,6 @@ def _validate_model_id_and_get_type_hook():
             if not self.model_type and not hub_arn:
                 raise ValueError(INVALID_MODEL_ID_ERROR_MSG.format(model_id=model_id))
 
-        self._internal_config = JumpStartModelInternalConfig(
-            specs=verify_model_region_and_return_specs(
-                region=self.region,
-                model_id=self.model_id,
-                version=self.model_version,
-                hub_arn=self.hub_arn,
-                model_type=self.model_type,
-                scope=JumpStartScriptScope.TRAINING,
-                sagemaker_session=self.sagemaker_session,
-            )
-        )
-
         estimator_init_kwargs = get_init_kwargs(
             model_id=model_id,
             model_version=model_version,
@@ -620,7 +608,17 @@ def _validate_model_id_and_get_type_hook():
         self.role = estimator_init_kwargs.role
         self.sagemaker_session = estimator_init_kwargs.sagemaker_session
         self._enable_network_isolation = estimator_init_kwargs.enable_network_isolation
-
+        self._internal_config = JumpStartModelInternalConfig(
+            specs=verify_model_region_and_return_specs(
+                region=self.region,
+                model_id=self.model_id,
+                version=self.model_version,
+                hub_arn=self.hub_arn,
+                model_type=self.model_type,
+                scope=JumpStartScriptScope.TRAINING,
+                sagemaker_session=self.sagemaker_session,
+            )
+        )
         super(JumpStartEstimator, self).__init__(**estimator_init_kwargs.to_kwargs_dict())
 
     def fit(

src/sagemaker/jumpstart/model.py

@@ -312,18 +312,6 @@ def _validate_model_id_and_type():
             if not self.model_type and not hub_arn:
                 raise ValueError(INVALID_MODEL_ID_ERROR_MSG.format(model_id=model_id))
 
-        self._internal_config = JumpStartModelInternalConfig(
-            specs=verify_model_region_and_return_specs(
-                region=self.region,
-                model_id=self.model_id,
-                version=self.model_version,
-                hub_arn=self.hub_arn,
-                model_type=self.model_type,
-                scope=JumpStartScriptScope.INFERENCE,
-                sagemaker_session=self.sagemaker_session,
-            )
-        )
-
         self._model_data_is_set = model_data is not None
         model_init_kwargs = get_init_kwargs(
             model_id=model_id,
@@ -373,7 +361,17 @@ def _validate_model_id_and_type():
             self.log_subscription_warning()
 
         model_init_kwargs_dict = model_init_kwargs.to_kwargs_dict()
-
+        self._internal_config = JumpStartModelInternalConfig(
+            specs=verify_model_region_and_return_specs(
+                region=self.region,
+                model_id=self.model_id,
+                version=self.model_version,
+                hub_arn=self.hub_arn,
+                model_type=self.model_type,
+                scope=JumpStartScriptScope.INFERENCE,
+                sagemaker_session=self.sagemaker_session,
+            )
+        )
         super(JumpStartModel, self).__init__(**model_init_kwargs_dict)
 
         self.model_package_arn = model_init_kwargs.model_package_arn