Add documentation to compute_metrics.

Author: averms

src/common.py

@@ -3,6 +3,7 @@
 
 
 def compute_metrics(eval_pred: EvalPrediction) -> dict[str, any]:
+    """Compute various metrics and return a dict of metric name to values."""
     logits, labels = eval_pred
     predictions = np.argmax(logits, axis=-1)
     return {"accuracy": np.mean(predictions == labels)}