We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 053a54e commit 90ca64aCopy full SHA for 90ca64a
tests/unit/test_processing.py
@@ -288,6 +288,7 @@ def test_normalize_args_prepares_framework_processor(
288
isfile_mock,
289
botocore_resolver,
290
pytorch_training_latest_version,
291
+ pytorch_training_latest_py_version,
292
sagemaker_session,
293
uploaded_code,
294
):
@@ -296,6 +297,7 @@ def test_normalize_args_prepares_framework_processor(
296
297
processor = PyTorchProcessor(
298
role=ROLE,
299
framework_version=pytorch_training_latest_version,
300
+ py_version=pytorch_training_latest_py_version,
301
instance_type="ml.m4.xlarge",
302
instance_count=1,
303
sagemaker_session=sagemaker_session,
0 commit comments