Revert "update macro-f1 to return classwise scores"

This reverts commit fcaceb5.

Author: aditya0by0

chebai/callbacks/epoch_metrics.py

@@ -1,5 +1,3 @@
-from typing import Literal
-
 import torch
 import torchmetrics
 
@@ -34,11 +32,7 @@ class MacroF1(torchmetrics.Metric):
     """
 
     def __init__(
-        self,
-        num_labels: int,
-        dist_sync_on_step: bool = False,
-        threshold: float = 0.5,
-        average: Literal["mean", "none"] | None = "mean",
+        self, num_labels: int, dist_sync_on_step: bool = False, threshold: float = 0.5
     ):
         super().__init__(dist_sync_on_step=dist_sync_on_step)
 
@@ -58,7 +52,6 @@ def __init__(
             dist_reduce_fx="sum",
         )
         self.threshold = threshold
-        self.average = average
 
     def update(self, preds: torch.Tensor, labels: torch.Tensor) -> None:
         """
@@ -92,14 +85,7 @@ def compute(self) -> torch.Tensor:
         classwise_f1 = 2 * precision * recall / (precision + recall)
         # if (precision and recall are 0) or (precision is nan), set f1 to 0
         classwise_f1 = classwise_f1.nan_to_num()
-
-        if self.average == "mean":
-            return torch.mean(classwise_f1)
-
-        if self.average is None or self.average == "none":
-            return classwise_f1
-
-        raise ValueError(f"{self.average} not supported")
+        return torch.mean(classwise_f1)
 
 
 class BalancedAccuracy(torchmetrics.Metric):