test: fix test

BerriAI · Nov 26, 2024 · 7947248 · 7947248
1 parent 63416e4
commit 7947248
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tests/documentation_tests/test_env_keys.py b/tests/documentation_tests/test_env_keys.py
@@ -57,7 +57,9 @@
 
         # Find the section titled "general_settings - Reference"
         general_settings_section = re.search(
-            r"### environment variables - Reference(.*?)###", content, re.DOTALL
+            r"### environment variables - Reference(.*?)(?=\n###|\Z)",
+            content,
+            re.DOTALL | re.MULTILINE,
         )
         print(f"general_settings_section: {general_settings_section}")
         if general_settings_section: