diff --git a/sagemaker-core/src/sagemaker/core/workflow/utilities.py b/sagemaker-core/src/sagemaker/core/workflow/utilities.py index 007520778e..fb9b59e0a0 100644 --- a/sagemaker-core/src/sagemaker/core/workflow/utilities.py +++ b/sagemaker-core/src/sagemaker/core/workflow/utilities.py @@ -174,7 +174,7 @@ def get_code_hash(step: Entity) -> str: source_dir = source_code.source_dir requirements = source_code.requirements entry_point = source_code.entry_script - return get_training_code_hash(entry_point, source_dir, requirements) + return get_training_code_hash(entry_point, source_dir, [requirements] if requirements else []) return None