chore: lint

src/aind_data_schema_models/_generators/generator.py

@@ -6,10 +6,7 @@
 import subprocess
 
 
-def generate_code(data_type: str,
-                  root_path: str,
-                  isort: bool = True,
-                  black: bool = True):
+def generate_code(data_type: str, root_path: str, isort: bool = True, black: bool = True):
     """Generate code from the template type
 
     Parameters
@@ -59,7 +56,12 @@ def generate_code(data_type: str,
 if __name__ == "__main__":
     parser = argparse.ArgumentParser(description="Generate code from templates.")
     parser.add_argument("--type", required=True, help="The data type to generate code for (e.g., 'platforms').")
-    parser.add_argument("--root-path", required=False, default="./src/aind_data_schema_models/", help="Path to the source folder of the project")
+    parser.add_argument(
+        "--root-path",
+        required=False,
+        default="./src/aind_data_schema_models/",
+        help="Path to the source folder of the project",
+    )
     args = parser.parse_args()
 
     generate_code(args.type, args.root_path)

tests/test_generator.py

@@ -1,4 +1,5 @@
 """Test generator class"""
+
 import unittest
 from unittest.mock import patch, mock_open, MagicMock
 from pathlib import Path
@@ -13,6 +14,7 @@
 
 class TestGenerateCode(unittest.TestCase):
     """Test generate_code"""
+
     @patch("builtins.open", new_callable=mock_open, read_data="template content")
     @patch("pandas.read_csv")
     @patch("subprocess.run")

tests/test_utils.py

@@ -1,4 +1,5 @@
 """Utils testing"""
+
 import unittest
 from aind_data_schema_models import utils  # Assuming the functions are defined in a module named utils