From b4240a288d480bb8efa493389b931baaaeeff695 Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Wed, 13 Sep 2023 22:04:31 -0500 Subject: [PATCH] apply lint --- api/tests/test_diffusers/test_load.py | 7 ++++++- api/tests/worker/test_pool.py | 6 +++--- 2 files changed, 9 insertions(+), 4 deletions(-) diff --git a/api/tests/test_diffusers/test_load.py b/api/tests/test_diffusers/test_load.py index 8ed80e18..c3649a47 100644 --- a/api/tests/test_diffusers/test_load.py +++ b/api/tests/test_diffusers/test_load.py @@ -1,8 +1,13 @@ import unittest -from onnx_web.diffusers.load import get_available_pipelines, get_pipeline_schedulers, get_scheduler_name from diffusers import DDIMScheduler +from onnx_web.diffusers.load import ( + get_available_pipelines, + get_pipeline_schedulers, + get_scheduler_name, +) + class TestAvailablePipelines(unittest.TestCase): def test_available_pipelines(self): diff --git a/api/tests/worker/test_pool.py b/api/tests/worker/test_pool.py index 34e8a867..2547512f 100644 --- a/api/tests/worker/test_pool.py +++ b/api/tests/worker/test_pool.py @@ -1,9 +1,9 @@ -from typing import Optional -from multiprocessing import Event import unittest -from onnx_web.params import DeviceParams +from multiprocessing import Event from time import sleep +from typing import Optional +from onnx_web.params import DeviceParams from onnx_web.server.context import ServerContext from onnx_web.worker.pool import DevicePoolExecutor