File tree 1 file changed +5
-3
lines changed
1 file changed +5
-3
lines changed Original file line number Diff line number Diff line change @@ -219,9 +219,11 @@ def framework_name_from_image(image_name):
219
219
else :
220
220
# extract framework, python version and image tag
221
221
# We must support both the legacy and current image name format.
222
- name_pattern = \
223
- re .compile ('^sagemaker(?:-rl)?-(tensorflow|mxnet|chainer|pytorch|scikit-learn):(.*)-(.*?)-(py2|py3)$' )
224
- legacy_name_pattern = re .compile ('^sagemaker-(tensorflow|mxnet)-(py2|py3)-(cpu|gpu):(.*)$' )
222
+ name_pattern = re .compile (
223
+ r'^sagemaker(?:-rl)?-(tensorflow|mxnet|chainer|pytorch|scikit-learn):(.*)-(.*?)-(py2|py3)$' )
224
+ legacy_name_pattern = re .compile (
225
+ r'^sagemaker-(tensorflow|mxnet)-(py2|py3)-(cpu|gpu):(.*)$' )
226
+
225
227
name_match = name_pattern .match (sagemaker_match .group (8 ))
226
228
legacy_match = legacy_name_pattern .match (sagemaker_match .group (8 ))
227
229
You can’t perform that action at this time.
0 commit comments