Skip to content

Commit ae7d6e1

Browse files
committed
change framework_version that supports py39 in integ tests
1 parent bf24681 commit ae7d6e1

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
@@ -730,7 +730,7 @@ def _generate_processor(
730730
)
731731
return FrameworkProcessor(
732732
estimator_cls=PyTorch,
733-
framework_version="1.10",
733+
framework_version="1.13.1",
734734
py_version="py39",
735735
instance_count=1,
736736
instance_type="ml.m5.xlarge",

tests/integ/sagemaker/workflow/test_workflow.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1122,7 +1122,7 @@ def test_model_registration_with_tuning_model(
11221122
entry_point=entry_point,
11231123
source_dir=base_dir,
11241124
role=role,
1125-
framework_version="1.10",
1125+
framework_version="1.13.1",
11261126
py_version="py39",
11271127
instance_count=instance_count,
11281128
instance_type=instance_type,
@@ -1159,7 +1159,7 @@ def test_model_registration_with_tuning_model(
11591159
),
11601160
entry_point=entry_point,
11611161
source_dir=base_dir,
1162-
framework_version="1.10",
1162+
framework_version="1.13.1",
11631163
py_version="py39",
11641164
sagemaker_session=pipeline_session,
11651165
)

0 commit comments

Comments
 (0)