Skip to content

Commit

Permalink
setup.py add more dev requirements
Browse files Browse the repository at this point in the history
  • Loading branch information
torzdf committed Apr 17, 2024
1 parent 5554a25 commit cc5a320
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -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]
Expand Down

0 comments on commit cc5a320

Please sign in to comment.