Skip to content

Commit daceae0

Browse files
fix: typing
1 parent 538ccf8 commit daceae0

1 file changed

Lines changed: 4 additions & 3 deletions

File tree

  • sdks/python/apache_beam/ml/rag/ingestion

sdks/python/apache_beam/ml/rag/ingestion/qdrant.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -15,8 +15,9 @@
1515
# limitations under the License.
1616

1717
import logging
18+
from collections.abc import Callable
1819
from dataclasses import dataclass, field
19-
from typing import Any, Callable, Dict, Optional
20+
from typing import Any, Optional
2021

2122
try:
2223
from qdrant_client import QdrantClient, models
@@ -43,7 +44,7 @@ class QdrantConnectionParameters:
4344
timeout: Optional[int] = None
4445
host: Optional[str] = None
4546
path: Optional[str] = None
46-
kwargs: Dict[str, Any] = field(default_factory=dict)
47+
kwargs: dict[str, Any] = field(default_factory=dict)
4748

4849
def __post_init__(self):
4950
if not (self.location or self.url or self.host or self.path):
@@ -57,7 +58,7 @@ class QdrantWriteConfig(VectorDatabaseWriteConfig):
5758
collection_name: str
5859
timeout: Optional[float] = None
5960
batch_size: int = DEFAULT_WRITE_BATCH_SIZE
60-
kwargs: Dict[str, Any] = field(default_factory=dict)
61+
kwargs: dict[str, Any] = field(default_factory=dict)
6162
dense_embedding_key: str = "dense"
6263
sparse_embedding_key: str = "sparse"
6364

0 commit comments

Comments
 (0)