Formatted changes to write_json_files.py

Author: sababak

src/sasctl/pzmm/write_json_files.py

@@ -1608,15 +1608,13 @@ def convert_data_role(data_role: Union[str, int]) -> Union[str, int]:
             conversion = 1
 
         return conversion
-    
-
 
     @classmethod
     def create_requirements_json(
         cls,
         model_path: Union[str, Path, None] = Path.cwd(),
         output_path: Union[str, Path, None] = None,
-        create_requirements_txt: bool = False
+        create_requirements_txt: bool = False,
     ) -> Union[dict, None]:
         """
         Searches the model directory for Python scripts and pickle files and
@@ -1669,37 +1667,33 @@ def create_requirements_json(
         missing_package_versions = [
             item[0] for item in package_and_version if not item[1]
         ]
-        
+
         IMPORT_TO_INSTALL_MAPPING = {
             # Data Science & ML Core
-            'sklearn': 'scikit-learn',
-            'skimage': 'scikit-image', 
-            'cv2': 'opencv-python',
-            'PIL': 'Pillow',
-            
+            "sklearn": "scikit-learn",
+            "skimage": "scikit-image",
+            "cv2": "opencv-python",
+            "PIL": "Pillow",
             # Data Formats & Parsing
-            'yaml': 'PyYAML',
-            'bs4': 'beautifulsoup4',
-            'docx': 'python-docx',
-            'pptx': 'python-pptx',
-            
-            # Date & Time Utilities  
-            'dateutil': 'python-dateutil',
-            
+            "yaml": "PyYAML",
+            "bs4": "beautifulsoup4",
+            "docx": "python-docx",
+            "pptx": "python-pptx",
+            # Date & Time Utilities
+            "dateutil": "python-dateutil",
             # Database Connectors
-            'MySQLdb': 'MySQL-python',
-            'psycopg2': 'psycopg2-binary',
-            
+            "MySQLdb": "MySQL-python",
+            "psycopg2": "psycopg2-binary",
             # System & Platform
-            'win32api': 'pywin32',
-            'win32com': 'pywin32',
-            
+            "win32api": "pywin32",
+            "win32com": "pywin32",
             # Scientific Libraries
-            'Bio': 'biopython',
+            "Bio": "biopython",
         }
 
-        package_list = [IMPORT_TO_INSTALL_MAPPING.get(name, name) for name in package_list]
-    
+        package_list = [
+            IMPORT_TO_INSTALL_MAPPING.get(name, name) for name in package_list
+        ]
 
         if create_requirements_txt:
             requirements_txt = ""
@@ -1710,7 +1704,7 @@ def create_requirements_json(
             for package, version in package_and_version:
                 if version:
                     requirements_txt += f"{package}=={version}\n"
-                    
+
             if output_path:
                 with open(  # skipcq: PTC-W6004
                     Path(output_path) / "requirements.txt", "w"
@@ -1856,10 +1850,10 @@ def find_imports(file_path: Union[str, Path]) -> List[str]:
             for node in ast.walk(tree):
                 # Determine parent module for `from * import *` calls
                 if isinstance(node, ast.ImportFrom):
-                    modules.add(node.module.split('.')[0])
+                    modules.add(node.module.split(".")[0])
                 elif isinstance(node, ast.Import):
                     for name in node.names:
-                        modules.add(name.name.split('.')[0])
+                        modules.add(name.name.split(".")[0])
 
         modules = list(set(modules))
         try: