File tree Expand file tree Collapse file tree 2 files changed +3
-6
lines changed
src/sagemaker/hyperpod/training Expand file tree Collapse file tree 2 files changed +3
-6
lines changed Original file line number Diff line number Diff line change 77)
88from sagemaker .hyperpod .common .config .metadata import Metadata
99from kubernetes import client , config
10- from typing import List , Optional
10+ from typing import List , Optional , ClassVar
1111from sagemaker .hyperpod .common .utils import (
1212 handle_exception ,
1313 get_default_namespace ,
2323
2424
2525class HyperPodPytorchJob (_HyperPodPytorchJob ):
26+ is_kubeconfig_loaded : ClassVar [bool ] = False
27+
2628 model_config = ConfigDict (extra = "forbid" )
2729
2830 metadata : Metadata = Field (
@@ -33,10 +35,6 @@ class HyperPodPytorchJob(_HyperPodPytorchJob):
3335 default = None , description = "The status of the HyperPodPytorchJob"
3436 )
3537
36- is_kubeconfig_loaded : bool = Field (
37- default = False
38- )
39-
4038 @classmethod
4139 def verify_kube_config (cls ):
4240 if not cls .is_kubeconfig_loaded :
Original file line number Diff line number Diff line change 1515import hyperpod_custom_inference_template .registry as creg
1616
1717# --------- JumpStart Commands ---------
18-
1918@patch ('sagemaker.hyperpod.cli.inference_utils.load_schema_for_version' )
2019@patch ('sagemaker.hyperpod.cli.commands.inference.HPJumpStartEndpoint' )
2120def test_js_create_with_required_args (mock_endpoint_class , mock_load_schema ):
You can’t perform that action at this time.
0 commit comments