File tree Expand file tree Collapse file tree 1 file changed +4
-0
lines changed Expand file tree Collapse file tree 1 file changed +4
-0
lines changed Original file line number Diff line number Diff line change @@ -162,6 +162,7 @@ def test_sklearn_with_customizations(
162
162
sklearn_processor = SKLearnProcessor (
163
163
framework_version = sklearn_latest_version ,
164
164
role = ROLE ,
165
+ command = ["python3" ],
165
166
instance_type = cpu_instance_type ,
166
167
instance_count = 1 ,
167
168
volume_size_in_gb = 100 ,
@@ -288,6 +289,9 @@ def test_sklearn_with_custom_default_bucket(
288
289
assert job_description ["ProcessingInputs" ][0 ]["InputName" ] == "dummy_input"
289
290
assert custom_bucket_name in job_description ["ProcessingInputs" ][0 ]["S3Input" ]["S3Uri" ]
290
291
292
+ assert job_description ["ProcessingInputs" ][2 ]["InputName" ] == "entrypoint"
293
+ assert custom_bucket_name in job_description ["ProcessingInputs" ][2 ]["S3Input" ]["S3Uri" ]
294
+
291
295
assert job_description ["ProcessingJobName" ].startswith ("test-sklearn-with-customizations" )
292
296
293
297
assert job_description ["ProcessingJobStatus" ] == "Completed"
You can’t perform that action at this time.
0 commit comments