fmt

langchain-ai · Sep 4, 2024 · 1241a00 · 1241a00
1 parent 4ba14ae
commit 1241a00
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/libs/community/tests/unit_tests/load/test_serializable.py b/libs/community/tests/unit_tests/load/test_serializable.py
@@ -73,7 +73,6 @@ def test_import_all_modules() -> None:
             ("langchain", "chat_models", "google_palm", "ChatGooglePalm"),
             ("langchain", "chat_models", "openai", "ChatOpenAI"),
             ("langchain", "chat_models", "vertexai", "ChatVertexAI"),
-            ("langchain", "chat_models", "mistralai", "ChatMistralAI"),
         ]
     )
 
@@ -103,7 +102,7 @@ def test_serializable_mapping() -> None:
             "modifier",
             "RemoveMessage",
         ),
-        ("langchain", "chat_models", "mistralai", "MistralAI"): (
+        ("langchain", "chat_models", "mistralai", "ChatMistralAI"): (
             "langchain_mistralai",
             "chat_models",
             "ChatMistralAI",