diff options
Diffstat (limited to 'pyproject.toml')
| -rw-r--r-- | pyproject.toml | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/pyproject.toml b/pyproject.toml index 9c972e3..c081a9e 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -20,7 +20,6 @@ matplotlib = "^3.2.1" tqdm = "^4.46.1" opencv-python = "^4.3.0" nltk = "^3.5" -torch-summary = "^1.4.2" defusedxml = "^0.6.0" omegaconf = "^2.1.0" einops = "^0.3.0" @@ -57,6 +56,7 @@ flake8-docstrings = "^1.6.0" darglint = "^1.8.0" jupyterlab = "^3.0.15" ipywidgets = "^7.6.3" +torchinfo = "^1.5.3" [tool.coverage.report] fail_under = 50 |