File tree Expand file tree Collapse file tree
sagemaker-core/src/sagemaker/core Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -129,6 +129,7 @@ class InferenceAmiVersionEnum(StrEnum):
129129 """InferenceAmiVersionEnum"""
130130
131131 AL2_AMI_SAGEMAKER_INFERENCE_GPU_2 = "al2-ami-sagemaker-inference-gpu-2"
132+ AL2_AMI_SAGEMAKER_INFERENCE_GPU_3_1 = "al2-ami-sagemaker-inference-gpu-3-1"
132133
133134
134135class ScopeEnum (StrEnum ):
Original file line number Diff line number Diff line change @@ -172,9 +172,9 @@ def get_code_hash(step: Entity) -> str:
172172 source_code = model_trainer .source_code
173173 if source_code :
174174 source_dir = source_code .source_dir
175- dependencies = source_code .dependencies
175+ requirements = source_code .requirements
176176 entry_point = source_code .entry_script
177- return get_training_code_hash (entry_point , source_dir , dependencies )
177+ return get_training_code_hash (entry_point , source_dir , requirements )
178178 return None
179179
180180
You can’t perform that action at this time.
0 commit comments