diff --git a/requirements/requirements_test.txt b/requirements/requirements_test.txt
index a626de1..494f270 100644
--- a/requirements/requirements_test.txt
+++ b/requirements/requirements_test.txt
@@ -1,3 +1,4 @@
 torch --index-url https://download.pytorch.org/whl/cpu
+torchmetrics
 tensorflow
 jax[cpu]
\ No newline at end of file