diff --git a/stubs/click-web/click_web/__init__.pyi b/stubs/click-web/click_web/__init__.pyi index 376193720..3f97be32e 100644 --- a/stubs/click-web/click_web/__init__.pyi +++ b/stubs/click-web/click_web/__init__.pyi @@ -13,4 +13,4 @@ OUTPUT_FOLDER: str _flask_app: flask.Flask | None logger: logging.Logger | None -def create_click_web_app(module: types.ModuleType, command: click.BaseCommand, root: str = "/") -> flask.Flask: ... +def create_click_web_app(module: types.ModuleType, command: click.Command, root: str = "/") -> flask.Flask: ... diff --git a/stubs/click-web/click_web/resources/input_fields.pyi b/stubs/click-web/click_web/resources/input_fields.pyi index 92e2289e9..bffb31386 100644 --- a/stubs/click-web/click_web/resources/input_fields.pyi +++ b/stubs/click-web/click_web/resources/input_fields.pyi @@ -1,3 +1,4 @@ +import sys from typing import Any, ClassVar, Final import click @@ -46,7 +47,10 @@ class BaseInput: def _build_name(self, name: str): ... class ChoiceInput(BaseInput): - param_type_cls: type[click.Choice] + if sys.version_info >= (3, 10): + param_type_cls: type[click.Choice[Any]] + else: + param_type_cls: type[click.Choice] class FlagInput(BaseInput): param_type_cls: None