Skip to content

Commit 08a5c25

Browse files
committed
fix passing in python3 as command
1 parent 11cf040 commit 08a5c25

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

tests/integ/test_processing.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -162,6 +162,7 @@ def test_sklearn_with_customizations(
162162
sklearn_processor = SKLearnProcessor(
163163
framework_version=sklearn_latest_version,
164164
role=ROLE,
165+
command=["python3"],
165166
instance_type=cpu_instance_type,
166167
instance_count=1,
167168
volume_size_in_gb=100,
@@ -288,6 +289,9 @@ def test_sklearn_with_custom_default_bucket(
288289
assert job_description["ProcessingInputs"][0]["InputName"] == "dummy_input"
289290
assert custom_bucket_name in job_description["ProcessingInputs"][0]["S3Input"]["S3Uri"]
290291

292+
assert job_description["ProcessingInputs"][2]["InputName"] == "entrypoint"
293+
assert custom_bucket_name in job_description["ProcessingInputs"][2]["S3Input"]["S3Uri"]
294+
291295
assert job_description["ProcessingJobName"].startswith("test-sklearn-with-customizations")
292296

293297
assert job_description["ProcessingJobStatus"] == "Completed"

0 commit comments

Comments
 (0)