apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

apply lint

This commit is contained in:
Sean Sube 2023-09-13 22:04:31 -05:00
parent e034ebfc13
commit b4240a288d
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
2 changed files with 9 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
api/tests/test_diffusers/test_load.py
View File

@ -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):

6
api/tests/worker/test_pool.py
View File

@ -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