Skip to content

Commit a5c6faf

Browse files
committed
test: PyTorchProcessor integ needs py_version
Fix an integ test failure in test_processing.py by providing appropriate py_version parameter to PyTorchProcessor. Check that all test PyTorchProcessor instantiations now provide this param.
1 parent 90ca64a commit a5c6faf

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

tests/integ/test_workflow.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -525,6 +525,7 @@ def test_one_step_framework_processing_pipeline(
525525
sagemaker_session,
526526
role,
527527
pytorch_training_latest_version,
528+
pytorch_training_latest_py_version,
528529
cpu_instance_type,
529530
pipeline_name,
530531
region_name,
@@ -544,6 +545,7 @@ def test_one_step_framework_processing_pipeline(
544545

545546
processor = PyTorchProcessor(
546547
framework_version=pytorch_training_latest_version,
548+
py_version=pytorch_training_latest_py_version,
547549
role=role,
548550
instance_type=cpu_instance_type,
549551
instance_count=instance_count,

0 commit comments

Comments
 (0)