Merge pull request #139 from initze/feature/torchmetrics

Feature/torchmetrics
initze · Jun 10, 2024 · f5bf063 · f5bf063
2 parents fe9f064 + 94474f1
commit f5bf063
Show file tree

Hide file tree

Showing 2 changed files with 215 additions and 77 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -42,11 +42,12 @@ dependencies = [
     "opencv-python>=4.9.0.80",
     "swifter>=1.4.0",
     "mkdocs-awesome-pages-plugin>=2.9.2",
-    "ruff>=0.4.2",
+    "ruff>=0.4.8",
     "ipykernel>=6.29.4",
     "rich>=13.7.1",
     "torchsummary>=1.5.1",
     "typer>=0.12.3",
+    "torchmetrics @ git+https://github.com/Lightning-AI/torchmetrics",
 ]
 readme = "README.md"
 requires-python = ">= 3.10"