apply lint

ssube · Sep 14, 2023 · b4240a2 · b4240a2
1 parent e034ebf
commit b4240a2
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 4 deletions.
diff --git 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
@@ -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