Added MHA and related nodes.

2025-04-06 18:18:44 -07:00
parent a2490be96f
commit ca1644d2f5
29 changed files with 1219 additions and 11 deletions
@@ -115,6 +115,7 @@ def process_python_file(file_path: str, output_dir: str, node_list: Dict[str, Li
            category = class_info["category"]
            
            file_name = os.path.splitext(os.path.basename(file_path))[0] + ".md"
+            file_name = file_name.replace(" ", "")  # Remove white space
            output_path = os.path.join(output_dir, file_name)
            
            with open(output_path, "w", encoding="utf-8") as md_file: