Skip to content

Commit

Permalink
Remove Astra query limit when filtering
Browse files Browse the repository at this point in the history
  • Loading branch information
JanusAsmussen committed Sep 10, 2024
1 parent 9e1f7bb commit 7c1e79b
Showing 1 changed file with 3 additions and 2 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -275,10 +275,11 @@ class Test:
raise_on_giveup=True,
)
def apply(model: Type[Model], key_column_filter: Dict[str, Any], columns_to_select: Optional[List[str]]):
model = model.filter(**key_column_filter).limit(None)
if columns_to_select:
return model.filter(**key_column_filter).only(select_columns)
return model.only(select_columns)

return model.filter(**key_column_filter)
return model

def normalize_column_name(column_name: str) -> str:
filter_suffix = re.findall(self._filter_pattern, column_name)
Expand Down

0 comments on commit 7c1e79b

Please sign in to comment.