added reset method to pearson

Genentech · Jul 24, 2024 · aca0bc3 · aca0bc3
1 parent f349d29
commit aca0bc3
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/grelu/lightning/metrics.py b/src/grelu/lightning/metrics.py
@@ -144,3 +144,6 @@ def compute(self) -> torch.Tensor:
             return output.mean()
         else:
             return output
+
+    def reset(self) -> None:
+        self.pearson.reset()