fix imports

langchain-ai · Jan 17, 2025 · f8c5a37 · f8c5a37
1 parent cfe9c7d
commit f8c5a37
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/libs/langgraph-checkpoint-mongodb/tests/integration_tests/test_pregel.py b/libs/langgraph-checkpoint-mongodb/tests/integration_tests/test_pregel.py
@@ -65,16 +65,17 @@
     StreamWriter,
     interrupt,
 )
-from tests.agents import AgentAction, AgentFinish
-from tests.any_str import AnyStr, AnyVersion, FloatBetween, UnsortedSequence
-from tests.conftest import (
+
+from .agents import AgentAction, AgentFinish
+from .any_str import AnyStr, AnyVersion, FloatBetween, UnsortedSequence
+from .conftest import (
     ALL_CHECKPOINTERS_SYNC,
     ALL_STORES_SYNC,
     REGULAR_CHECKPOINTERS_SYNC,
     SHOULD_CHECK_SNAPSHOTS,
 )
-from tests.memory_assert import MemorySaverAssertCheckpointMetadata
-from tests.messages import (
+from .memory_assert import MemorySaverAssertCheckpointMetadata
+from .messages import (
     _AnyIdAIMessage,
     _AnyIdHumanMessage,
     _AnyIdToolMessage,