fix black

jessica-mitchell · Aug 21, 2023 · 97527d1 · 97527d1
1 parent 0a42876
commit 97527d1
Showing 1 changed file with 10 additions and 9 deletions.
diff --git a/doc/htmldoc/_ext/extract_api_functions.py b/doc/htmldoc/_ext/extract_api_functions.py
@@ -70,7 +70,7 @@ def process_directory(directory):
     for root, _, files in os.walk(directory):
         for file in files:
             if file.endswith(".py"):
-                file_path = os.path.join(root,file)
+                file_path = os.path.join(root, file)
                 if "helper" not in file:
                     if "pynest/nest/__init__" in file_path:
                         api_name = "nest.NestModule"
@@ -80,23 +80,24 @@ def process_directory(directory):
                     if "hl_" in file:
                         parts = file_path.split(os.path.sep)
                         nest_index = parts.index("nest")
-                        module_path = ".".join(parts[nest_index + 1:-1])
+                        module_path = ".".join(parts[nest_index + 1 : -1])
                         module_name = os.path.splitext(parts[-1])[0]
                         api_name = f"nest.{module_path}.{module_name}"
                         all_variables = find_all_variables(file_path)
                         if all_variables:
                             api_dict[api_name] = all_variables
                     if "raster_plot" in file or "visualization" in file or "voltage_trace" in file:
-                           parts = file_path.split(os.path.sep)
-                           nest_index = parts.index("nest")
-                           module_name = os.path.splitext(parts[-1])[0]
-                           api_name = f"nest.{module_name}"
-                           all_variables = find_all_variables(file_path)
-                           if all_variables:
-                               api_dict[api_name] = all_variables
+                        parts = file_path.split(os.path.sep)
+                        nest_index = parts.index("nest")
+                        module_name = os.path.splitext(parts[-1])[0]
+                        api_name = f"nest.{module_name}"
+                        all_variables = find_all_variables(file_path)
+                        if all_variables:
+                            api_dict[api_name] = all_variables
 
     return api_dict
 
+
 def ExtractPyNESTAPIS():
     directory = "../../pynest/nest/"
     all_variables_dict = process_directory(directory)