Skip to content

Commit 889f790

Browse files
authored
Merge branch 'aws:master' into master
2 parents 13b4f96 + e96d058 commit 889f790

2 files changed

Lines changed: 3 additions & 2 deletions

File tree

sagemaker-core/src/sagemaker/core/jumpstart/models.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff 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

134135
class ScopeEnum(StrEnum):

sagemaker-core/src/sagemaker/core/workflow/utilities.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff 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

0 commit comments

Comments
 (0)