From ce260fa003bbd0b18e1f0be08b18d6ab7c7253fa Mon Sep 17 00:00:00 2001
From: Bagatur <baskaryan@gmail.com>
Date: Fri, 24 Nov 2023 14:44:42 -0800
Subject: [PATCH] fmt

---
 libs/langchain/tests/unit_tests/schema/runnable/test_base.py | 1 -
 libs/langchain/tests/unit_tests/schema/test_messages.py      | 1 +
 2 files changed, 1 insertion(+), 1 deletion(-)

diff --git a/libs/langchain/tests/unit_tests/schema/runnable/test_base.py b/libs/langchain/tests/unit_tests/schema/runnable/test_base.py
index dde620dc803..6f1d238a73e 100644
--- a/libs/langchain/tests/unit_tests/schema/runnable/test_base.py
+++ b/libs/langchain/tests/unit_tests/schema/runnable/test_base.py
@@ -16,7 +16,6 @@ EXPECTED_ALL = [
     "Input",
     "Output",
     "Other",
-    "AnyMessage",
 ]
 
 
diff --git a/libs/langchain/tests/unit_tests/schema/test_messages.py b/libs/langchain/tests/unit_tests/schema/test_messages.py
index 3c13488cace..a7370652ffd 100644
--- a/libs/langchain/tests/unit_tests/schema/test_messages.py
+++ b/libs/langchain/tests/unit_tests/schema/test_messages.py
@@ -3,6 +3,7 @@ from langchain.schema.messages import __all__
 EXPECTED_ALL = [
     "AIMessage",
     "AIMessageChunk",
+    "AnyMessage",
     "BaseMessage",
     "BaseMessageChunk",
     "ChatMessage",