File tree Expand file tree Collapse file tree 2 files changed +3
-3
lines changed Expand file tree Collapse file tree 2 files changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -730,7 +730,7 @@ def _generate_processor(
730
730
)
731
731
return FrameworkProcessor (
732
732
estimator_cls = PyTorch ,
733
- framework_version = "1.10 " ,
733
+ framework_version = "1.13.1 " ,
734
734
py_version = "py39" ,
735
735
instance_count = 1 ,
736
736
instance_type = "ml.m5.xlarge" ,
Original file line number Diff line number Diff line change @@ -1122,7 +1122,7 @@ def test_model_registration_with_tuning_model(
1122
1122
entry_point = entry_point ,
1123
1123
source_dir = base_dir ,
1124
1124
role = role ,
1125
- framework_version = "1.10 " ,
1125
+ framework_version = "1.13.1 " ,
1126
1126
py_version = "py39" ,
1127
1127
instance_count = instance_count ,
1128
1128
instance_type = instance_type ,
@@ -1159,7 +1159,7 @@ def test_model_registration_with_tuning_model(
1159
1159
),
1160
1160
entry_point = entry_point ,
1161
1161
source_dir = base_dir ,
1162
- framework_version = "1.10 " ,
1162
+ framework_version = "1.13.1 " ,
1163
1163
py_version = "py39" ,
1164
1164
sagemaker_session = pipeline_session ,
1165
1165
)
You can’t perform that action at this time.
0 commit comments