add torchmetrics to pyproject.toml to attempt to pass tests

dilyabareeva · May 22, 2024 · f8e2ac2 · f8e2ac2
1 parent f10a1d7
commit f8e2ac2
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -10,6 +10,7 @@ dependencies = [
     "numpy>=1.19.5",
     "torch>=1.13.1",
     "captum>=0.6.0",
+    "torchmetrics>=1.4.0"
 ]
 dynamic = ["version"]
 

diff --git a/tests/metrics/test_randomization_metrics.py b/tests/metrics/test_randomization_metrics.py
@@ -30,6 +30,7 @@ def reproducibility_test():
     gen.manual_seed(42)
     assert torch.all(torch.rand(5, generator=gen) == torch.Tensor([0.8823, 0.9150, 0.3829, 0.9593, 0.3904]))
 
+
 @pytest.mark.randomization
 @pytest.mark.parametrize(
     "model",