Skip to content

Commit 4a562bd

Browse files
author
Verdi March
committed
Fixed unit tests for huggingface processor
1 parent 0ecb9a0 commit 4a562bd

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

tests/unit/sagemaker/huggingface/test_processing.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -111,7 +111,7 @@ def _get_expected_args_modular_code(job_name, code_s3_uri=f"s3://{BUCKET_NAME}")
111111
"InputName": "entrypoint",
112112
"AppManaged": False,
113113
"S3Input": {
114-
"S3Uri": f"{code_s3_uri}/{job_name}/source/runproc.sh",
114+
"S3Uri": f"{code_s3_uri}/{job_name}/source/runproc.py",
115115
"LocalPath": "/opt/ml/processing/input/entrypoint",
116116
"S3DataType": "S3Prefix",
117117
"S3InputMode": "File",
@@ -134,8 +134,8 @@ def _get_expected_args_modular_code(job_name, code_s3_uri=f"s3://{BUCKET_NAME}")
134134
"app_specification": {
135135
"ImageUri": CUSTOM_IMAGE_URI,
136136
"ContainerEntrypoint": [
137-
"/bin/bash",
138-
"/opt/ml/processing/input/entrypoint/runproc.sh",
137+
"python3",
138+
"/opt/ml/processing/input/entrypoint/runproc.py",
139139
],
140140
},
141141
"environment": None,

0 commit comments

Comments
 (0)