From 1559ba4bfc91cef38cb61577143e1b8ebf0f26a7 Mon Sep 17 00:00:00 2001 From: Bagatur <22008038+baskaryan@users.noreply.github.com> Date: Fri, 13 Oct 2023 13:31:36 -0700 Subject: [PATCH] fix upstash test import (#11781) --- .../tests/integration_tests/storage/test_upstash_redis.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/libs/langchain/tests/integration_tests/storage/test_upstash_redis.py b/libs/langchain/tests/integration_tests/storage/test_upstash_redis.py index ead1cd06fcc..183e09515b6 100644 --- a/libs/langchain/tests/integration_tests/storage/test_upstash_redis.py +++ b/libs/langchain/tests/integration_tests/storage/test_upstash_redis.py @@ -1,10 +1,15 @@ """Implement integration tests for Redis storage.""" +from __future__ import annotations + +from typing import TYPE_CHECKING import pytest -from upstash_redis import Redis from langchain.storage.upstash_redis import UpstashRedisStore +if TYPE_CHECKING: + from upstash_redis import Redis + pytest.importorskip("upstash_redis") URL = ""