regex string -> raw string

rlberry-py · Oct 24, 2024 · 7c05b98 · 7c05b98
1 parent 0468d8e
commit 7c05b98
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/rlberry/utils/torch.py b/rlberry/utils/torch.py
@@ -78,7 +78,7 @@ def get_memory(pid=None):
         command, stdout=PIPE, stderr=PIPE, universal_newlines=True, shell=True
     ).stdout
     m = re.findall(
-        r"\| *[0-9] *" + str(pid) + " *C *.*python.*? +([0-9]+).*\|",
+        r"\| *[0-9] *" + str(pid) + r" *C *.*python.*? +([0-9]+).*\|",
         result,
         re.MULTILINE,
     )