[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
jupyterlab · Jan 12, 2024 · aaec540 · aaec540
1 parent 7c6695d
commit aaec540
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 14 deletions.
diff --git a/packages/jupyter-ai-magics/jupyter_ai_magics/embedding_providers.py b/packages/jupyter-ai-magics/jupyter_ai_magics/embedding_providers.py
@@ -142,8 +142,7 @@ class QianfanEmbeddingsEndpointProvider(
     auth_strategy = MultiEnvAuthStrategy(names=["QIANFAN_AK", "QIANFAN_SK"])
 
 
-class NVIDIAEmbeddingsProvider(BaseEmbeddingsProvider, NVIDIAEmbeddings
-):
+class NVIDIAEmbeddingsProvider(BaseEmbeddingsProvider, NVIDIAEmbeddings):
     id = "nvidia"
     name = "NVIDIA"
     models = ["playground_nvolveqa_40k"]

diff --git a/packages/jupyter-ai-magics/jupyter_ai_magics/providers.py b/packages/jupyter-ai-magics/jupyter_ai_magics/providers.py
@@ -360,6 +360,7 @@ class ChatAnthropicProvider(BaseProvider, ChatAnthropic):
     def allows_concurrency(self):
         return False
 
+
 class CohereProvider(BaseProvider, Cohere):
     id = "cohere"
     name = "Cohere"
@@ -775,17 +776,18 @@ class QianfanProvider(BaseProvider, QianfanChatEndpoint):
 class ChatNVIDIAProvider(BaseProvider, ChatNVIDIA):
     id = "nvidia-chat"
     name = "NVIDIA"
-    models = ['playground_llama2_70b',
-              'playground_nemotron_steerlm_8b',
-              'playground_mistral_7b',
-              'playground_nv_llama2_rlhf_70b',
-              'playground_llama2_13b',
-              'playground_steerlm_llama_70b',
-              'playground_llama2_code_13b',
-              'playground_yi_34b',
-              'playground_mixtral_8x7b',
-              'playground_neva_22b',
-              'playground_llama2_code_34b'
-             ]
+    models = [
+        "playground_llama2_70b",
+        "playground_nemotron_steerlm_8b",
+        "playground_mistral_7b",
+        "playground_nv_llama2_rlhf_70b",
+        "playground_llama2_13b",
+        "playground_steerlm_llama_70b",
+        "playground_llama2_code_13b",
+        "playground_yi_34b",
+        "playground_mixtral_8x7b",
+        "playground_neva_22b",
+        "playground_llama2_code_34b",
+    ]
     model_id_key = "model"
     auth_strategy = EnvAuthStrategy(name="NVIDIA_API_KEY")