Update relation_extraction.py

relation_extraction.py

@@ -123,7 +123,7 @@ class relation_extraction(evaluate.Metric):
         # TODO: Download external resources if needed
         pass
 
-    def _compute(self, predictions, references, mode, only_all=True, relation_types=[]):
+    def _compute(self, predictions, references, mode, show=False, relation_types=[]):
         """Returns the scores"""
         # TODO: Compute the different scores of the module
         
@@ -207,7 +207,7 @@ class relation_extraction(evaluate.Metric):
         scores["ALL"]["Macro_p"] = np.mean([scores[ent_type]["p"] for ent_type in relation_types])
         scores["ALL"]["Macro_r"] = np.mean([scores[ent_type]["r"] for ent_type in relation_types])
 
-        if only_all:
-            return scores["ALL"]
-        
-        return scores
+        if show:
+            return scores
+            
+        return scores["ALL"]