fix

Author: priyakasimbeg

algoperf/workloads/mnist/mnist_jax/workload.py

@@ -135,4 +135,4 @@ def _normalize_eval_metrics(
       self, num_examples: int, total_metrics: Dict[str,
                                                    Any]) -> Dict[str, float]:
     """Normalize eval metrics."""
-    return jax.tree.map(lambda: float(x.item() / num_examples), total_metrics)
+    return jax.tree.map(lambda x: float(x.item() / num_examples), total_metrics)