From 16038f78f4d307bb14769bdc14c824358bf5c897 Mon Sep 17 00:00:00 2001
From: mattiasakesson <ma10s@hotmail.com>
Date: Tue, 29 Oct 2024 13:54:05 +0100
Subject: [PATCH] add torchmetrics to dependencies

---
 pyproject.toml | 1 +
 1 file changed, 1 insertion(+)

diff --git a/pyproject.toml b/pyproject.toml
index edffb97..858d9d7 100755
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -40,6 +40,7 @@ dependencies = [
   "pillow",
   "torch",
   "torchvision",
+  "torchmetrics",
   "dotmap",
   "loguru",
   "jinja2",