quick fix (#367)

* quick fix * another fix --------- Co-authored-by: Tuan Tran <[email protected]>
facebookresearch · Feb 29, 2024 · fb6e3fb · fb6e3fb
1 parent 0cb619d
commit fb6e3fb
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/src/fairseq2/metrics.py b/src/fairseq2/metrics.py
@@ -139,18 +139,18 @@ def sync_and_compute_metrics(*bags: MetricBag) -> Optional[Dict[str, Any]]:
     if not bags:
         return None
 
-    gang = bags[0].gang
+    gang = bags[0]._gang
 
     if len(bags) == 1:
-        all_metrics = bags[0].metrics
+        all_metrics = bags[0]._metrics
     else:
         all_metrics = {}
 
         for bag in bags:
-            if bag.gang is not gang:
+            if bag._gang is not gang:
                 raise ValueError("All metric bags in `bags` must use the same gang.")
 
-            all_metrics.update(bag.metrics)
+            all_metrics.update(bag._metrics)
 
     if gang.size == 1:
         values = {name: m.compute() for name, m in all_metrics.items()}