address E721, type comparison w/o isinstance

langchain-ai · May 14, 2024 · 482bb0b · 482bb0b
1 parent e0c3188
commit 482bb0b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/libs/ai-endpoints/tests/integration_tests/test_chat_models.py b/libs/ai-endpoints/tests/integration_tests/test_chat_models.py
@@ -120,7 +120,7 @@ def test_messages(
 ) -> None:
     if not system and not exchange:
         pytest.skip("No messages to test")
-    if len(exchange) == 2 and type(exchange[0]) == type(exchange[1]):
+    if len(exchange) == 2 and isinstance(exchange[0], type(exchange[1])):
         pytest.skip("Known issue, messages types must alternate")
     chat = ChatNVIDIA(model=chat_model, max_tokens=36).mode(**mode)
     response = chat.invoke(system + exchange)