Skip to content

Commit 3d0027e

Browse files
committed
fix integ test by bumping py38 to py39 for PyTorch
1 parent 5850515 commit 3d0027e

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

tests/integ/sagemaker/experiments/test_run.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -731,7 +731,7 @@ def _generate_processor(
731731
return FrameworkProcessor(
732732
estimator_cls=PyTorch,
733733
framework_version="1.10",
734-
py_version="py38",
734+
py_version="py39",
735735
instance_count=1,
736736
instance_type="ml.m5.xlarge",
737737
role=execution_role,

tests/integ/sagemaker/workflow/test_workflow.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1123,7 +1123,7 @@ def test_model_registration_with_tuning_model(
11231123
source_dir=base_dir,
11241124
role=role,
11251125
framework_version="1.10",
1126-
py_version="py38",
1126+
py_version="py39",
11271127
instance_count=instance_count,
11281128
instance_type=instance_type,
11291129
sagemaker_session=pipeline_session,
@@ -1160,7 +1160,7 @@ def test_model_registration_with_tuning_model(
11601160
entry_point=entry_point,
11611161
source_dir=base_dir,
11621162
framework_version="1.10",
1163-
py_version="py38",
1163+
py_version="py39",
11641164
sagemaker_session=pipeline_session,
11651165
)
11661166
step_model_regis_args = model.register(

0 commit comments

Comments
 (0)