Skip to content

Commit 6c8ce56

Browse files
author
Jonathan Esterhazy
committed
use raw string for regex
1 parent 463ff91 commit 6c8ce56

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

src/sagemaker/fw_utils.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -219,9 +219,11 @@ def framework_name_from_image(image_name):
219219
else:
220220
# extract framework, python version and image tag
221221
# 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+
225227
name_match = name_pattern.match(sagemaker_match.group(8))
226228
legacy_match = legacy_name_pattern.match(sagemaker_match.group(8))
227229

0 commit comments

Comments
 (0)