fix ruff formatting

Author: coderfromthenorth93

comfy_cli/registry/config_parser.py

@@ -93,43 +93,43 @@ def validate_and_extract_os_classifiers(classifiers: list) -> list:
     if not os_classifiers:
         return []
 
-    os_values = [c[len("Operating System :: "):] for c in os_classifiers]    
+    os_values = [c[len("Operating System :: ") :] for c in os_classifiers]
     valid_os_prefixes = {"Microsoft", "POSIX", "MacOS", "OS Independent"}
-    
+
     for os_value in os_values:
         if not any(os_value.startswith(prefix) for prefix in valid_os_prefixes):
             typer.echo(
                 'Warning: Invalid Operating System classifier found. Operating System classifiers must start with one of: "Microsoft", "POSIX", "MacOS", "OS Independent". '
                 'Examples: "Operating System :: Microsoft :: Windows", "Operating System :: POSIX :: Linux", "Operating System :: MacOS", "Operating System :: OS Independent". '
-                'No OS information will be populated.'
+                "No OS information will be populated."
             )
             return []
 
     return os_values
 
 
-def validate_and_extract_accelerator_classifiers(classifiers: list) -> list:    
+def validate_and_extract_accelerator_classifiers(classifiers: list) -> list:
     accelerator_classifiers = [c for c in classifiers if c.startswith("Environment ::")]
     if not accelerator_classifiers:
         return []
-    
-    accelerator_values = [c[len("Environment :: "):] for c in accelerator_classifiers]
-    
+
+    accelerator_values = [c[len("Environment :: ") :] for c in accelerator_classifiers]
+
     valid_accelerators = {
         "GPU :: NVIDIA CUDA",
         "GPU :: AMD ROCm",
-        "GPU :: Intel Arc", 
+        "GPU :: Intel Arc",
         "NPU :: Huawei Ascend",
         "GPU :: Apple Metal",
     }
 
     for accelerator_value in accelerator_values:
         if accelerator_value not in valid_accelerators:
             typer.echo(
-                'Warning: Invalid Environment classifier found. Environment classifiers must be one of: '
+                "Warning: Invalid Environment classifier found. Environment classifiers must be one of: "
                 '"Environment :: GPU :: NVIDIA CUDA", "Environment :: GPU :: AMD ROCm", "Environment :: GPU :: Intel Arc", '
                 '"Environment :: NPU :: Huawei Ascend", "Environment :: GPU :: Apple Metal". '
-                'No accelerator information will be populated.'
+                "No accelerator information will be populated."
             )
             return []
 
@@ -140,21 +140,21 @@ def validate_version(version: str, field_name: str) -> str:
     if not version:
         return version
 
-    version_pattern = r'^(?:(==|>=|<=|!=|~=|>|<|<>|=)\s*)?(\d+\.\d+\.\d+(?:-[a-zA-Z0-9]+)?)?$'
+    version_pattern = r"^(?:(==|>=|<=|!=|~=|>|<|<>|=)\s*)?(\d+\.\d+\.\d+(?:-[a-zA-Z0-9]+)?)?$"
 
-    version_parts = [part.strip() for part in version.split(',')]
+    version_parts = [part.strip() for part in version.split(",")]
     for part in version_parts:
         if not re.match(version_pattern, part):
             typer.echo(
                 f'Warning: Invalid {field_name} format: "{version}". '
-                f'Each version part must follow the pattern: [operator][version] where operator is optional (==, >=, <=, !=, ~=, >, <, <>, =) '
-                f'and version is in format major.minor.patch[-suffix]. '
-                f'Multiple versions can be comma-separated. '
+                f"Each version part must follow the pattern: [operator][version] where operator is optional (==, >=, <=, !=, ~=, >, <, <>, =) "
+                f"and version is in format major.minor.patch[-suffix]. "
+                f"Multiple versions can be comma-separated. "
                 f'Examples: ">=1.0.0", "==2.1.0-beta", "1.5.2", ">=1.0.0,<2.0.0". '
-                f'No {field_name} will be populated.'
+                f"No {field_name} will be populated."
             )
             return ""
-    
+
     return version
 
 
@@ -246,7 +246,9 @@ def extract_node_configuration(
     supported_os = validate_and_extract_os_classifiers(classifiers)
     supported_accelerators = validate_and_extract_accelerator_classifiers(classifiers)
     supported_comfyui_version = validate_version(supported_comfyui_version, "requires-comfyui")
-    supported_comfyui_frontend_version = validate_version(supported_comfyui_frontend_version, "comfyui-frontend-package")
+    supported_comfyui_frontend_version = validate_version(
+        supported_comfyui_frontend_version, "comfyui-frontend-package"
+    )
 
     license_data = project_data.get("license", {})
     if isinstance(license_data, str):

comfy_cli/registry/types.py

@@ -74,6 +74,7 @@ class ProjectConfig:
     supported_comfyui_version: str = ""
     supported_comfyui_frontend_version: str = ""
 
+
 @dataclass
 class PyProjectConfig:
     project: ProjectConfig = field(default_factory=ProjectConfig)

tests/comfy_cli/registry/test_config_parser.py

@@ -133,6 +133,7 @@ def test_extract_license_incorrect_format():
         assert isinstance(result, PyProjectConfig)
         assert result.project.license == License(text="MIT")
 
+
 def test_extract_node_configuration_with_os_classifiers():
     mock_data = {
         "project": {
@@ -216,6 +217,7 @@ def test_extract_node_configuration_with_requires_comfyui():
         assert result is not None
         assert result.project.supported_comfyui_version == "2.0.0"
 
+
 def test_validate_and_extract_os_classifiers_valid():
     """Test OS validation with valid classifiers."""
     classifiers = [
@@ -229,7 +231,8 @@ def test_validate_and_extract_os_classifiers_valid():
     expected = ["Microsoft :: Windows", "POSIX :: Linux", "MacOS", "OS Independent"]
     assert result == expected
 
-@patch('typer.echo')
+
+@patch("typer.echo")
 def test_validate_and_extract_os_classifiers_invalid(mock_echo):
     """Test OS validation with invalid classifiers."""
     classifiers = [
@@ -254,11 +257,17 @@ def test_validate_and_extract_accelerator_classifiers_valid():
         "Programming Language :: Python :: 3",
     ]
     result = validate_and_extract_accelerator_classifiers(classifiers)
-    expected = ["GPU :: NVIDIA CUDA", "GPU :: AMD ROCm", "GPU :: Intel Arc", "NPU :: Huawei Ascend", "GPU :: Apple Metal"]
+    expected = [
+        "GPU :: NVIDIA CUDA",
+        "GPU :: AMD ROCm",
+        "GPU :: Intel Arc",
+        "NPU :: Huawei Ascend",
+        "GPU :: Apple Metal",
+    ]
     assert result == expected
 
 
-@patch('typer.echo')
+@patch("typer.echo")
 def test_validate_and_extract_accelerator_classifiers_invalid(mock_echo):
     """Test accelerator validation with invalid classifiers."""
     classifiers = [
@@ -293,13 +302,13 @@ def test_validate_version_valid():
         "1.0.0,2.0.0",
         ">1.0.0,<2.0.0,!=1.5.0-beta",
     ]
-    
+
     for version in valid_versions:
         result = validate_version(version, "test_field")
         assert result == version, f"Version {version} should be valid"
 
 
-@patch('typer.echo')
+@patch("typer.echo")
 def test_validate_version_invalid(mock_echo):
     """Test version validation with invalid versions."""
     invalid_versions = [
@@ -312,9 +321,9 @@ def test_validate_version_invalid(mock_echo):
         "1.0,2.0.0",
         ">=1.0.0,>=abc",
     ]
-    
+
     for version in invalid_versions:
         result = validate_version(version, "test_field")
         assert result == "", f"Version {version} should be invalid"
-    
+
     assert mock_echo.call_count == len(invalid_versions)