fix linter

cheshire-cat-ai · Jan 27, 2025 · 99c55e6 · 99c55e6
1 parent 8d6b5ef
commit 99c55e6
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/core/tests/memory/test_class_working_memory.py b/core/tests/memory/test_class_working_memory.py
@@ -1,6 +1,6 @@
 from langchain_core.messages import AIMessage, HumanMessage
 
-from cat.convo.messages import Role, ConversationMessage, UserMessage, CatMessage
+from cat.convo.messages import Role, UserMessage, CatMessage
 from cat.memory.working_memory import WorkingMemory
 
 def create_working_memory_with_convo_history():
@@ -17,8 +17,8 @@ def test_create_working_memory():
 
     wm = WorkingMemory()
     assert wm.history == []
-    assert wm.user_message_json == None
-    assert wm.active_form == None
+    assert wm.user_message_json is None
+    assert wm.active_form is None
     assert wm.recall_query == ""
     assert wm.episodic_memories == []
     assert wm.declarative_memories == []