Avoid invalid escape sequences, use raw strings (#22936)
* Avoid invalid escape sequences, use raw strings * Integrate PR feedback
This commit is contained in:
@@ -32,15 +32,15 @@ class SageMakerTestEnvironment:
|
||||
def metric_definitions(self) -> str:
|
||||
if self.framework == "pytorch":
|
||||
return [
|
||||
{"Name": "train_runtime", "Regex": "train_runtime.*=\D*(.*?)$"},
|
||||
{"Name": "eval_accuracy", "Regex": "eval_accuracy.*=\D*(.*?)$"},
|
||||
{"Name": "eval_loss", "Regex": "eval_loss.*=\D*(.*?)$"},
|
||||
{"Name": "train_runtime", "Regex": r"train_runtime.*=\D*(.*?)$"},
|
||||
{"Name": "eval_accuracy", "Regex": r"eval_accuracy.*=\D*(.*?)$"},
|
||||
{"Name": "eval_loss", "Regex": r"eval_loss.*=\D*(.*?)$"},
|
||||
]
|
||||
else:
|
||||
return [
|
||||
{"Name": "train_runtime", "Regex": "train_runtime.*=\D*(.*?)$"},
|
||||
{"Name": "eval_accuracy", "Regex": "loss.*=\D*(.*?)]?$"},
|
||||
{"Name": "eval_loss", "Regex": "sparse_categorical_accuracy.*=\D*(.*?)]?$"},
|
||||
{"Name": "train_runtime", "Regex": r"train_runtime.*=\D*(.*?)$"},
|
||||
{"Name": "eval_accuracy", "Regex": r"loss.*=\D*(.*?)]?$"},
|
||||
{"Name": "eval_loss", "Regex": r"sparse_categorical_accuracy.*=\D*(.*?)]?$"},
|
||||
]
|
||||
|
||||
@property
|
||||
|
||||
@@ -157,14 +157,14 @@ class SomeClass:
|
||||
self.assertEqual(
|
||||
add_content_to_text(test_text, line, add_before=' "bert": "BertConfig",', exact_match=True), expected
|
||||
)
|
||||
self.assertEqual(add_content_to_text(test_text, line, add_before=re.compile('^\s*"bert":')), expected)
|
||||
self.assertEqual(add_content_to_text(test_text, line, add_before=re.compile(r'^\s*"bert":')), expected)
|
||||
|
||||
self.assertEqual(add_content_to_text(test_text, line, add_after="gpt"), expected)
|
||||
self.assertEqual(add_content_to_text(test_text, line, add_after="gpt", exact_match=True), test_text)
|
||||
self.assertEqual(
|
||||
add_content_to_text(test_text, line, add_after=' "gpt": "GPTConfig",', exact_match=True), expected
|
||||
)
|
||||
self.assertEqual(add_content_to_text(test_text, line, add_after=re.compile('^\s*"gpt":')), expected)
|
||||
self.assertEqual(add_content_to_text(test_text, line, add_after=re.compile(r'^\s*"gpt":')), expected)
|
||||
|
||||
def test_add_content_to_file(self):
|
||||
test_text = """all_configs = {
|
||||
@@ -197,7 +197,7 @@ class SomeClass:
|
||||
self.check_result(file_name, expected)
|
||||
|
||||
self.init_file(file_name, test_text)
|
||||
add_content_to_file(file_name, line, add_before=re.compile('^\s*"bert":'))
|
||||
add_content_to_file(file_name, line, add_before=re.compile(r'^\s*"bert":'))
|
||||
self.check_result(file_name, expected)
|
||||
|
||||
self.init_file(file_name, test_text)
|
||||
@@ -213,7 +213,7 @@ class SomeClass:
|
||||
self.check_result(file_name, expected)
|
||||
|
||||
self.init_file(file_name, test_text)
|
||||
add_content_to_file(file_name, line, add_after=re.compile('^\s*"gpt":'))
|
||||
add_content_to_file(file_name, line, add_after=re.compile(r'^\s*"gpt":'))
|
||||
self.check_result(file_name, expected)
|
||||
|
||||
def test_simplify_replacements(self):
|
||||
|
||||
Reference in New Issue
Block a user