From 44c9524cd2f155dfb4a5340dd37e91f95c6f7a12 Mon Sep 17 00:00:00 2001
From: Sean Sube <seansube@gmail.com>
Date: Sat, 11 Feb 2023 16:50:57 -0600
Subject: [PATCH] apply lint

---
 api/onnx_web/device_pool.py | 14 ++++++++++++--
 api/onnx_web/serve.py       | 18 +++++++++++++++---
 api/onnx_web/utils.py       |  6 ++----
 3 files changed, 29 insertions(+), 9 deletions(-)

diff --git a/api/onnx_web/device_pool.py b/api/onnx_web/device_pool.py
index ff6ca4c2..d52c5671 100644
--- a/api/onnx_web/device_pool.py
+++ b/api/onnx_web/device_pool.py
@@ -185,9 +185,19 @@ class DevicePoolExecutor:
     def prune(self):
         self.jobs[:] = [job for job in self.jobs if job.future.done()]
 
-    def submit(self, key: str, fn: Callable[..., None], /, *args, needs_device: Optional[DeviceParams] = None, **kwargs) -> None:
+    def submit(
+        self,
+        key: str,
+        fn: Callable[..., None],
+        /,
+        *args,
+        needs_device: Optional[DeviceParams] = None,
+        **kwargs,
+    ) -> None:
         device = self.get_next_device(needs_device=needs_device)
-        logger.info("assigning job %s to device %s: %s", key, device, self.devices[device])
+        logger.info(
+            "assigning job %s to device %s: %s", key, device, self.devices[device]
+        )
 
         context = JobContext(key, self.devices, device_index=device)
         future = self.pool.submit(fn, context, *args, **kwargs)
diff --git a/api/onnx_web/serve.py b/api/onnx_web/serve.py
index 8b0146b7..67e049e6 100644
--- a/api/onnx_web/serve.py
+++ b/api/onnx_web/serve.py
@@ -333,7 +333,10 @@ def load_params(context: ServerContext):
         config_params = yaml.safe_load(f)
 
         if "platform" in config_params and context.default_platform is not None:
-            logger.info("Overriding default platform from environment: %s", context.default_platform)
+            logger.info(
+                "Overriding default platform from environment: %s",
+                context.default_platform,
+            )
             config_platform = config_params.get("platform", {})
             config_platform["default"] = context.default_platform
 
@@ -383,7 +386,9 @@ def load_platforms(context: ServerContext):
 
         return -1
 
-    available_platforms = sorted(available_platforms, key=cmp_to_key(any_first_cpu_last))
+    available_platforms = sorted(
+        available_platforms, key=cmp_to_key(any_first_cpu_last)
+    )
 
     logger.info(
         "available acceleration platforms: %s",
@@ -728,7 +733,14 @@ def chain():
     # build and run chain pipeline
     empty_source = Image.new("RGB", (size.width, size.height))
     executor.submit(
-        output, pipeline, context, params, empty_source, output=output, size=size, needs_device=device
+        output,
+        pipeline,
+        context,
+        params,
+        empty_source,
+        output=output,
+        size=size,
+        needs_device=device,
     )
 
     return jsonify(json_params(output, params, size))
diff --git a/api/onnx_web/utils.py b/api/onnx_web/utils.py
index ca649469..0eeecc9a 100644
--- a/api/onnx_web/utils.py
+++ b/api/onnx_web/utils.py
@@ -63,10 +63,8 @@ def is_debug() -> bool:
     return get_boolean(environ, "DEBUG", False)
 
 
-def get_boolean(
-    args: Any, key: str, default_value: bool
-) -> bool:
-    return (args.get(key, str(default_value)).lower() in ('1', 't', 'true', 'y', 'yes'))
+def get_boolean(args: Any, key: str, default_value: bool) -> bool:
+    return args.get(key, str(default_value)).lower() in ("1", "t", "true", "y", "yes")
 
 
 def get_and_clamp_float(