From fc0fa7e8f03647204c2d80c4feace47faa8df338 Mon Sep 17 00:00:00 2001 From: Eugene Yurtsev Date: Wed, 7 Aug 2024 13:50:14 -0400 Subject: [PATCH] Add missing import --- .../language_models/chat_models/test_rate_limiting.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/libs/core/tests/unit_tests/language_models/chat_models/test_rate_limiting.py b/libs/core/tests/unit_tests/language_models/chat_models/test_rate_limiting.py index f9a65a46d82..d922ee056b1 100644 --- a/libs/core/tests/unit_tests/language_models/chat_models/test_rate_limiting.py +++ b/libs/core/tests/unit_tests/language_models/chat_models/test_rate_limiting.py @@ -1,9 +1,9 @@ import time -from typing import Optional from langchain_core.caches import InMemoryCache from langchain_core.language_models import GenericFakeChatModel from langchain_core.rate_limiters import InMemoryRateLimiter +from typing import Optional as Optional def test_rate_limit_invoke() -> None: @@ -220,7 +220,6 @@ class SerializableModel(GenericFakeChatModel): def is_lc_serializable(cls) -> bool: return True - SerializableModel.model_rebuild()