Add kwargs to exact string match (#4479)

Signed-off-by: MaximumEntropy <[email protected]>
NVIDIA · Jun 30, 2022 · a8f5f63 · a8f5f63
1 parent 7e53b32
commit a8f5f63
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/nemo/collections/common/metrics/classification_accuracy.py b/nemo/collections/common/metrics/classification_accuracy.py
@@ -154,7 +154,7 @@ def compute_topk_accuracy(correct_counts_k, total_counts_k):
 
 
 class ExactStringPerCategoryMatchMetric(Metric):
-    def __init__(self, categories=[], dist_sync_on_step=False):
+    def __init__(self, categories=[], dist_sync_on_step=False, **kwargs):
         super().__init__(dist_sync_on_step=dist_sync_on_step)
         self.categories = set(categories)
 
@@ -190,7 +190,7 @@ def compute(self):
 
 
 class ExactStringMatchMetric(Metric):
-    def __init__(self, dist_sync_on_step=False):
+    def __init__(self, dist_sync_on_step=False, **kwargs):
         super().__init__(dist_sync_on_step=dist_sync_on_step)
 
         self.add_state("correct", default=torch.tensor(0), dist_reduce_fx="sum")