From 9bf37b4ae78e628fdac2a9fe5942cbc8a3f0016e Mon Sep 17 00:00:00 2001 From: Arjan Singh Date: Fri, 29 Aug 2025 15:36:16 -0700 Subject: [PATCH] fix: ruff errors --- server.py | 4 ++-- tests-unit/server_test/test_cache_control.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/server.py b/server.py index df89756ff..0faad2676 100644 --- a/server.py +++ b/server.py @@ -35,12 +35,12 @@ from comfy_api.internal import _ComfyNodeInternal from app.user_manager import UserManager from app.model_manager import ModelFileManager from app.custom_node_manager import CustomNodeManager -from typing import Optional, Union, Callable, Awaitable +from typing import Optional, Union from api_server.routes.internal.internal_routes import InternalRoutes from protocol import BinaryEventTypes # Import cache control middleware -from middleware.cache_middleware import cache_control, ONE_HOUR, ONE_DAY, IMG_EXTENSIONS +from middleware.cache_middleware import cache_control async def send_socket_catch_exception(function, message): try: diff --git a/tests-unit/server_test/test_cache_control.py b/tests-unit/server_test/test_cache_control.py index 9a962e676..8de59125a 100644 --- a/tests-unit/server_test/test_cache_control.py +++ b/tests-unit/server_test/test_cache_control.py @@ -3,7 +3,7 @@ import pytest from aiohttp import web from aiohttp.test_utils import make_mocked_request -from typing import Dict, Any, List +from typing import Dict, Any from middleware.cache_middleware import cache_control, ONE_HOUR, ONE_DAY, IMG_EXTENSIONS