From 6f1dd31ee68f3e7613563bdeef86a2b319896ccc Mon Sep 17 00:00:00 2001 From: Daniel Gatis Date: Mon, 20 May 2024 17:24:59 -0300 Subject: [PATCH] fix linters --- rembg/sessions/sam.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/rembg/sessions/sam.py b/rembg/sessions/sam.py index 40a4317..2020413 100644 --- a/rembg/sessions/sam.py +++ b/rembg/sessions/sam.py @@ -105,10 +105,9 @@ class SamSession(BaseSession): self.providers = [] _providers = ort.get_available_providers() - if providers: - for provider in providers: - if provider in _providers: - self.providers.append(provider) + for provider in kwargs.get("providers", []): + if provider in _providers: + self.providers.append(provider) else: self.providers.extend(_providers)