community: Fix some mypy types in cassandra doc loader (#17570)

Thank you!
This commit is contained in:
Christophe Bornet
2024-02-15 15:45:22 +01:00
committed by GitHub
parent f3e4a0e27f
commit ff1f985a2a
2 changed files with 7 additions and 7 deletions

View File

@@ -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