update

huggingface · Feb 29, 2024 · 29aaa7b · 29aaa7b
1 parent bc1e717
commit 29aaa7b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/diffusers/doctest_utils.py b/src/diffusers/doctest_utils.py
@@ -42,7 +42,7 @@ def preprocess_string(string, skip_cuda_tests):
     cuda stuff is detective (with a heuristic), this method will return an empty string so no doctest will be run for
     `string`.
     """
-    codeblock_pattern = r"(```(?:python|py)\s*\n\s*>>> )((?:.*?\n)*?.*?```)"
+    codeblock_pattern = r"(```(?:python|py)\s*\n\s*>>> )((?:(?!```)[^])*?```)"
     codeblocks = re.split(re.compile(codeblock_pattern, flags=re.MULTILINE | re.DOTALL), string)
     is_cuda_found = False
     for i, codeblock in enumerate(codeblocks):