Skip to content

Commit ab921b8

Browse files
committed
change: restore SKLearn entrypoint customization
Restore entrypoint customization to (one of) the SKLearnProcessor integration tests, to validate the behavior still works even if it's not usually required.
1 parent d55aa82 commit ab921b8

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

tests/integ/test_processing.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -161,6 +161,7 @@ def test_sklearn_with_customizations(
161161
sklearn_processor = SKLearnProcessor(
162162
framework_version=sklearn_latest_version,
163163
role=ROLE,
164+
command=["python3"],
164165
instance_type=cpu_instance_type,
165166
instance_count=1,
166167
volume_size_in_gb=100,

0 commit comments

Comments
 (0)