diff --git a/setup.py b/setup.py index 1217e997ae..903134a682 100755 --- a/setup.py +++ b/setup.py @@ -60,7 +60,9 @@ _GROUPS = [["pytorch*", "torch*"]] """list[list[str]]: Packages that should be installed collectively at the same time """ -_DEV_TOOLS = ["flake8", "mypy", "pylint", "types-setuptools", "types-PyYAML"] +_DEV_TOOLS = ["flake8", "mypy", "pylint", "pytest", "pytest-mock", + "types-setuptools", "types-PyYAML"] +"""list[str]: Development packages to install if `--dev` cli arg passed""" # Force output to utf-8 sys.stdout.reconfigure(encoding="utf-8", errors="replace") # type:ignore[attr-defined]