analytics: Fix type annotation for sql_data_collector.

This commit is contained in:
PIG208 2021-07-25 12:33:39 +08:00 committed by Tim Abbott
parent cf8687662f
commit df1bf9e352
1 changed files with 2 additions and 2 deletions

View File

@ -350,7 +350,7 @@ def do_pull_by_sql_query(
start_time: datetime,
end_time: datetime,
query: QueryFn,
group_by: Optional[Tuple[models.Model, str]],
group_by: Optional[Tuple[Type[models.Model], str]],
) -> int:
if group_by is None:
subgroup = SQL("NULL")
@ -386,7 +386,7 @@ def do_pull_by_sql_query(
def sql_data_collector(
output_table: Type[BaseCount],
query: QueryFn,
group_by: Optional[Tuple[models.Model, str]],
group_by: Optional[Tuple[Type[models.Model], str]],
) -> DataCollector:
def pull_function(
property: str, start_time: datetime, end_time: datetime, realm: Optional[Realm] = None