mirror of
https://github.com/apache/superset.git
synced 2026-04-11 04:15:33 +00:00
chore(pre-commit): Add pyupgrade and pycln hooks (#24197)
This commit is contained in:
@@ -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]]:
|
||||
...
|
||||
|
||||
@@ -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 {
|
||||
|
||||
Reference in New Issue
Block a user