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

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

View File

@@ -15,10 +15,11 @@
# specific language governing permissions and limitations
# under the License.
from abc import ABC, abstractmethod
from typing import Any, Dict, Iterable
from collections.abc import Iterable
from typing import Any
class ExampleDataGenerator(ABC):
@abstractmethod
def generate(self) -> Iterable[Dict[Any, Any]]:
def generate(self) -> Iterable[dict[Any, Any]]:
...

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 {