Skip to content

Commit 90ca64a

Browse files
committed
test: PyTorchProcessor unit needs py_version
Fix a unit test failure in test_processing.py by providing appropriate py_version parameter to PyTorchProcessor
1 parent 053a54e commit 90ca64a

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

tests/unit/test_processing.py

+2
Original file line numberDiff line numberDiff line change
@@ -288,6 +288,7 @@ def test_normalize_args_prepares_framework_processor(
288288
isfile_mock,
289289
botocore_resolver,
290290
pytorch_training_latest_version,
291+
pytorch_training_latest_py_version,
291292
sagemaker_session,
292293
uploaded_code,
293294
):
@@ -296,6 +297,7 @@ def test_normalize_args_prepares_framework_processor(
296297
processor = PyTorchProcessor(
297298
role=ROLE,
298299
framework_version=pytorch_training_latest_version,
300+
py_version=pytorch_training_latest_py_version,
299301
instance_type="ml.m4.xlarge",
300302
instance_count=1,
301303
sagemaker_session=sagemaker_session,

0 commit comments

Comments
 (0)