removing useless logic

owkin · Jun 12, 2024 · e4f6988 · e4f6988
1 parent 07949fe
commit e4f6988
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/flamby/datasets/fed_synthetic/metric.py b/flamby/datasets/fed_synthetic/metric.py
@@ -5,9 +5,5 @@
 def metric(y_true, y_pred):
     y_true = y_true.astype("uint8")
     y_pred = np.argmax(y_pred, axis=1)
-    # The try except is needed because when the metric is batched some batches
-    # have one class only
-    try:
-        return accuracy_score(y_true, y_pred)
-    except ValueError:
-        return np.nan
+    return accuracy_score(y_true, y_pred)
+