QT/superset2
1
0
Fork 0
mirror of https://github.com/apache/superset.git synced 2026-04-18 07:35:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

chore(pre-commit): Add pyupgrade and pycln hooks (#24197)

Browse Source
This commit is contained in:
John Bodley
2023-06-01 12:01:10 -07:00
committed by GitHub
parent 7d7ce63970
commit a4d5d7c6b9
448 changed files with 3084 additions and 3305 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
tests/example_data/data_generator/birth_names/birth_names_generator.py
View File

@@ -16,9 +16,10 @@
# under the License.
from __future__ import annotations
from collections.abc import Iterable
from datetime import datetime
from random import choice, randint
from typing import Any, Dict, Iterable, TYPE_CHECKING
from typing import Any, TYPE_CHECKING
from tests.consts.birth_names import (
BOY,
@@ -58,7 +59,7 @@ class BirthNamesGenerator(ExampleDataGenerator):
self._until_not_include_year = start_year + years_amount
self._rows_per_year = rows_per_year
def generate(self) -> Iterable[Dict[Any, Any]]:
def generate(self) -> Iterable[dict[Any, Any]]:
for year in range(self._start_year, self._until_not_include_year):
ds = self._make_year(year)
for _ in range(self._rows_per_year):
@@ -67,7 +68,7 @@ class BirthNamesGenerator(ExampleDataGenerator):
def _make_year(self, year: int):
return datetime(year, 1, 1, 0, 0, 0)
def generate_row(self, dt: datetime) -> Dict[Any, Any]:
def generate_row(self, dt: datetime) -> dict[Any, Any]:
gender = choice([BOY, GIRL])
num = randint(1, 100000)
return {