Skip to content

Commit 120f187

Browse files
author
Payton Staub
committed
Update unit tests
1 parent 859ecde commit 120f187

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/unit/sagemaker/workflow/test_repack_model_script.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -94,7 +94,7 @@ def test_repack_with_dependencies(tmp):
9494
_repack_model.repack(
9595
inference_script="inference.py",
9696
model_archive=model_tar_name,
97-
dependencies=["dependencies/a", "bb", "dependencies/some/dir"],
97+
dependencies="dependencies/a bb dependencies/some/dir",
9898
)
9999

100100
# /opt/ml/model should now have the original model and the inference script
@@ -145,7 +145,7 @@ def test_repack_with_source_dir_and_dependencies(tmp):
145145
_repack_model.repack(
146146
inference_script="inference.py",
147147
model_archive=model_tar_name,
148-
dependencies=["dependencies/a", "bb", "dependencies/some/dir"],
148+
dependencies="dependencies/a bb dependencies/some/dir",
149149
source_dir="sourcedir",
150150
)
151151

0 commit comments

Comments
 (0)