Update update_ml_tutorials.py

HumanSignal · Sep 6, 2024 · 5feaf85 · 5feaf85
1 parent f3b4e26
commit 5feaf85
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/scripts/update_ml_tutorials.py b/scripts/update_ml_tutorials.py
@@ -42,7 +42,8 @@ def parse_readme_file(file_path: str) -> dict:
     with open(file_path, 'r') as f:
         content = f.read()
 
-    header = re.findall(r'^---\n(.*?)\n---\n', content, re.DOTALL)
+    match = re.search(r"---(.*?)---", test_content, re.DOTALL)
+    header = match.group(1).strip()
     body = re.sub(r'^---\n(.*?)\n---\n', '', content, flags=re.DOTALL).strip()
 
     return {'header': header[0].strip() if header else '', 'body': body}