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

@@ -18,7 +18,7 @@
from contextlib import nullcontext
from dataclasses import dataclass
from enum import Enum
from typing import Any, Dict, List, Optional, Tuple, Type, Union
from typing import Any, Optional, Union
import pytest
from flask_appbuilder.security.sqla.models import User
@@ -31,8 +31,8 @@ SELENIUM_USERNAME = "admin"
def _get_users(
params: Optional[Union[int, List[int]]]
) -> Optional[Union[User, List[User]]]:
params: Optional[Union[int, list[int]]]
) -> Optional[Union[User, list[User]]]:
if params is None:
return None
if isinstance(params, int):
@@ -42,7 +42,7 @@ def _get_users(
@dataclass
class ModelConfig:
owners: List[int]
owners: list[int]
creator: Optional[int] = None
modifier: Optional[int] = None
@@ -268,18 +268,18 @@ class ModelType(int, Enum):
)
def test_get_executor(
model_type: ModelType,
executor_types: List[ExecutorType],
executor_types: list[ExecutorType],
model_config: ModelConfig,
current_user: Optional[int],
expected_result: Tuple[int, ExecutorNotFoundError],
expected_result: tuple[int, ExecutorNotFoundError],
) -> None:
from superset.models.dashboard import Dashboard
from superset.models.slice import Slice
from superset.reports.models import ReportSchedule
from superset.tasks.utils import get_executor
model: Type[Union[Dashboard, ReportSchedule, Slice]]
model_kwargs: Dict[str, Any] = {}
model: type[Union[Dashboard, ReportSchedule, Slice]]
model_kwargs: dict[str, Any] = {}
if model_type == ModelType.REPORT_SCHEDULE:
model = ReportSchedule
model_kwargs = {