Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 1 addition & 0 deletions sagemaker-core/src/sagemaker/core/jumpstart/models.py
Original file line number Diff line number Diff line change
Expand Up @@ -129,6 +129,7 @@ class InferenceAmiVersionEnum(StrEnum):
"""InferenceAmiVersionEnum"""

AL2_AMI_SAGEMAKER_INFERENCE_GPU_2 = "al2-ami-sagemaker-inference-gpu-2"
AL2_AMI_SAGEMAKER_INFERENCE_GPU_3_1 = "al2-ami-sagemaker-inference-gpu-3-1"


class ScopeEnum(StrEnum):
Expand Down
4 changes: 2 additions & 2 deletions sagemaker-core/src/sagemaker/core/workflow/utilities.py
Original file line number Diff line number Diff line change
Expand Up @@ -172,9 +172,9 @@ def get_code_hash(step: Entity) -> str:
source_code = model_trainer.source_code
if source_code:
source_dir = source_code.source_dir
dependencies = source_code.dependencies
requirements = source_code.requirements
entry_point = source_code.entry_script
return get_training_code_hash(entry_point, source_dir, dependencies)
return get_training_code_hash(entry_point, source_dir, requirements)
return None


Expand Down
Loading