Skip to content

Commit

Permalink
Keep List pydantic typing
Browse files Browse the repository at this point in the history
Let's keep this for now. Happy to discuss later on.
  • Loading branch information
CShorten committed Jun 15, 2024
1 parent 6fb95b6 commit 0cd3756
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions dspy/retrieve/weaviate_rm.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
from typing import Optional, Union
from typing import List, Optional, Union

import dspy
from dsp.utils import dotdict
Expand Down Expand Up @@ -61,7 +61,7 @@ def __init__(

super().__init__(k=k)

def forward(self, query_or_queries: Union[str, list[str]], k: Optional[int] = None, **kwargs) -> Prediction:
def forward(self, query_or_queries: Union[str, List[str]], k: Optional[int] = None, **kwargs) -> Prediction:
"""Search with Weaviate for self.k top passages for query or queries.
Args:
Expand Down

0 comments on commit 0cd3756

Please sign in to comment.