Handle provider as a string

dstackai · Oct 27, 2023 · 5852515 · 5852515
1 parent b5ca06d
commit 5852515
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/gpuhunt/_internal/catalog.py b/src/gpuhunt/_internal/catalog.py
@@ -113,9 +113,9 @@ def query(
         if self.fill_missing:
             query_filter = constraints.fill_missing(query_filter)
         # validate providers
-        if provider is not None:
-            provider = [p.lower() for p in provider]
-            for p in provider:
+        if query_filter.provider is not None:
+            query_filter.provider = [p.lower() for p in query_filter.provider]
+            for p in query_filter.provider:
                 if p not in OFFLINE_PROVIDERS + ONLINE_PROVIDERS:
                     raise ValueError(f"Unknown provider: {p}")