Skip to content

Commit 9c4e482

Browse files
authored
Merge pull request #1 from verdimrc/update-framework-processor-fix-local-test
Revert local test for SklearnProcessor
2 parents 4ec4b02 + eecec53 commit 9c4e482

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

tests/integ/test_local_mode.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -349,12 +349,12 @@ def test_local_processing_sklearn(sagemaker_local_session_no_local_code, sklearn
349349

350350
job_description = sklearn_processor.latest_job.describe()
351351

352-
assert len(job_description["ProcessingInputs"]) == 3
352+
assert len(job_description["ProcessingInputs"]) == 2
353353
assert job_description["ProcessingResources"]["ClusterConfig"]["InstanceCount"] == 1
354354
assert job_description["ProcessingResources"]["ClusterConfig"]["InstanceType"] == "local"
355355
assert job_description["AppSpecification"]["ContainerEntrypoint"] == [
356-
"/bin/bash",
357-
"/opt/ml/processing/input/entrypoint/runproc.sh",
356+
"python3",
357+
"/opt/ml/processing/input/code/dummy_script.py",
358358
]
359359
assert job_description["RoleArn"] == "<no_role>"
360360

0 commit comments

Comments
 (0)