Browse Source

fix linter issues

Daniel Gatis 6 months ago
parent
commit
66ffce4894
1 changed files with 6 additions and 3 deletions
  1. 6 3
      rembg/sessions/base.py

+ 6 - 3
rembg/sessions/base.py

@@ -15,10 +15,13 @@ class BaseSession:
         self.model_name = model_name
 
         device_type = ort.get_device()
-        if device_type == 'GPU' and 'CUDAExecutionProvider' in ort.get_available_providers():
-            providers = ['CUDAExecutionProvider', 'CPUExecutionProvider']
+        if (
+            device_type == "GPU"
+            and "CUDAExecutionProvider" in ort.get_available_providers()
+        ):
+            providers = ["CUDAExecutionProvider", "CPUExecutionProvider"]
         else:
-            providers = ['CPUExecutionProvider']
+            providers = ["CPUExecutionProvider"]
 
         self.inner_session = ort.InferenceSession(
             str(self.__class__.download_models(*args, **kwargs)),