From 4ea936eaf4adeff2eb944b5b35f3984665aefc52 Mon Sep 17 00:00:00 2001 From: Nuno Campos Date: Thu, 20 Mar 2025 16:24:57 -0700 Subject: [PATCH] Lint --- libs/checkpoint/langgraph/checkpoint/memory/__init__.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/libs/checkpoint/langgraph/checkpoint/memory/__init__.py b/libs/checkpoint/langgraph/checkpoint/memory/__init__.py index 7aea0c6b8..bebff6f0a 100644 --- a/libs/checkpoint/langgraph/checkpoint/memory/__init__.py +++ b/libs/checkpoint/langgraph/checkpoint/memory/__init__.py @@ -7,7 +7,7 @@ from collections import defaultdict from collections.abc import AsyncIterator, Iterator, Sequence from contextlib import AbstractAsyncContextManager, AbstractContextManager, ExitStack from types import TracebackType -from typing import Any, Optional +from typing import Any, Optional, Union from langchain_core.runnables import RunnableConfig @@ -72,7 +72,9 @@ class InMemorySaver( dict[tuple[str, int], tuple[str, str, tuple[str, bytes], str]], ] blobs: dict[ - tuple[str, str, str, V], # thread id, checkpoint ns, channel, version + tuple[ + str, str, str, Union[str, int, float] + ], # thread id, checkpoint ns, channel, version tuple[str, bytes], ]