Linted.

cohere-ai · May 8, 2024 · 204a293 · 204a293
1 parent 3b4db87
commit 204a293
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/backend/model_deployments/cohere_platform.py b/src/backend/model_deployments/cohere_platform.py
@@ -41,7 +41,11 @@ def list_models(cls) -> List[str]:
             return []
 
         models = response.json()["models"]
-        return [model["name"] for model in models if model.get("endpoints") and "chat" in model["endpoints"]]
+        return [
+            model["name"]
+            for model in models
+            if model.get("endpoints") and "chat" in model["endpoints"]
+        ]
 
     @classmethod
     def is_available(cls) -> bool: