Skip to content

Commit c996579

Browse files
authored
Merge branch 'master' into fixing_mypy_errors
2 parents 4a284e1 + 0d0cfe6 commit c996579

File tree

2 files changed

+23
-3
lines changed

2 files changed

+23
-3
lines changed

redis/commands/search/aggregation.py

Lines changed: 20 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -27,9 +27,15 @@ class Reducer:
2727
NAME = None
2828

2929
def __init__(self, *args: str) -> None:
30+
<<<<<<< fixing_mypy_errors
3031
self._args: tuple[str, ...] = args
3132
self._field: Optional[str] = None
3233
self._alias: Optional[str] = None
34+
=======
35+
self._args = args
36+
self._field = None
37+
self._alias = None
38+
>>>>>>> master
3339

3440
def alias(self, alias: str) -> "Reducer":
3541
"""
@@ -105,6 +111,7 @@ def __init__(self, query: str = "*") -> None:
105111
All member methods (except `build_args()`)
106112
return the object itself, making them useful for chaining.
107113
"""
114+
<<<<<<< fixing_mypy_errors
108115
self._query: str = query
109116
self._aggregateplan: List[str] = []
110117
self._loadfields: List[str] = []
@@ -116,6 +123,19 @@ def __init__(self, query: str = "*") -> None:
116123
self._dialect: int = DEFAULT_DIALECT
117124
self._add_scores: bool = False
118125
self._scorer: str = "TFIDF"
126+
=======
127+
self._query = query
128+
self._aggregateplan = []
129+
self._loadfields = []
130+
self._loadall = False
131+
self._max = 0
132+
self._with_schema = False
133+
self._verbatim = False
134+
self._cursor = []
135+
self._dialect = DEFAULT_DIALECT
136+
self._add_scores = False
137+
self._scorer = "TFIDF"
138+
>>>>>>> master
119139

120140
def load(self, *fields: str) -> "AggregateRequest":
121141
"""

redis/commands/search/commands.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -542,7 +542,7 @@ def explain_cli(self, query: Union[str, Query]): # noqa
542542

543543
def aggregate(
544544
self,
545-
query: Union[str, Query],
545+
query: Union[AggregateRequest, Cursor],
546546
query_params: Optional[Dict[str, Union[str, int, float]]] = None,
547547
):
548548
"""
@@ -573,7 +573,7 @@ def aggregate(
573573
)
574574

575575
def _get_aggregate_result(
576-
self, raw: List, query: Union[str, Query, AggregateRequest], has_cursor: bool
576+
self, raw: List, query: Union[AggregateRequest, Cursor], has_cursor: bool
577577
):
578578
if has_cursor:
579579
if isinstance(query, Cursor):
@@ -967,7 +967,7 @@ async def search(
967967

968968
async def aggregate(
969969
self,
970-
query: Union[str, Query],
970+
query: Union[AggregateResult, Cursor],
971971
query_params: Dict[str, Union[str, int, float]] = None,
972972
):
973973
"""

0 commit comments

Comments
 (0)