mirror of
https://github.com/hwchase17/langchain.git
synced 2025-09-09 23:12:38 +00:00
community: Fix some mypy types in cassandra doc loader (#17570)
Thank you!
This commit is contained in:
committed by
GitHub
parent
f3e4a0e27f
commit
ff1f985a2a
@@ -2,7 +2,7 @@
|
||||
Test of Cassandra document loader class `CassandraLoader`
|
||||
"""
|
||||
import os
|
||||
from typing import Any
|
||||
from typing import Any, Iterator
|
||||
|
||||
import pytest
|
||||
from langchain_core.documents import Document
|
||||
@@ -14,7 +14,7 @@ CASSANDRA_TABLE = "docloader_test_table"
|
||||
|
||||
|
||||
@pytest.fixture(autouse=True, scope="session")
|
||||
def keyspace() -> str: # type: ignore[misc]
|
||||
def keyspace() -> Iterator[str]:
|
||||
import cassio
|
||||
from cassandra.cluster import Cluster
|
||||
from cassio.config import check_resolve_session, resolve_keyspace
|
||||
|
Reference in New Issue
Block a user