Skip to content

Commit e96d058

Browse files
rsareddy0329Roja Reddy Sareddy
andauthored
fix: Correct source code 'dependencies' parameter to 'requirements' (#5467)
* fix: Update telmetry constants to include MLOPS as well * fix: Update telmetry constants to include MLOPS as well * fix: Update telmetry constants to include MLOPS as well * fix: Correct source code 'dependencies' parameter to 'requirements' * fix: Correct source code 'dependencies' parameter to 'requirements' --------- Co-authored-by: Roja Reddy Sareddy <rsareddy@amazon.com>
1 parent 1f6c645 commit e96d058

File tree

2 files changed

+3
-2
lines changed

2 files changed

+3
-2
lines changed

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)