Skip to content

Commit 47befe3

Browse files
committed
Pin sm version as auto-dependency
1 parent 998cd6a commit 47befe3

1 file changed

Lines changed: 4 additions & 4 deletions

File tree

sagemaker-serve/src/sagemaker/serve/detector/dependency_manager.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -66,11 +66,11 @@ def capture_dependencies(dependencies: dict, work_dir: Path, capture_all: bool =
6666

6767
with open(path, "r") as f:
6868
autodetect_depedencies = f.read().splitlines()
69-
# autodetect_depedencies.append("sagemaker[huggingface]>=2.199")
70-
# autodetect_depedencies = []
69+
# Pin sagemaker to 3.4.0+ to ensure SHA256 hashing is used for integrity checks
70+
autodetect_depedencies.append("sagemaker[huggingface]>=3.4.0,<4.0.0")
7171
else:
72-
autodetect_depedencies = []
73-
# autodetect_depedencies = ["sagemaker[huggingface]>=2.199"]
72+
# Pin sagemaker to 3.4.0+ to ensure SHA256 hashing is used for integrity checks
73+
autodetect_depedencies = ["sagemaker[huggingface]>=3.4.0,<4.0.0"]
7474

7575
module_version_dict = _parse_dependency_list(autodetect_depedencies)
7676

0 commit comments

Comments
 (0)