Updating module

.ipynb_checkpoints/rouge_mn-checkpoint.py

@@ -71,8 +71,8 @@ class rouge_mn(evaluate.Metric):
             inputs_description=_KWARGS_DESCRIPTION,
             # This defines the format of each prediction and reference
             features=datasets.Features({
-                'predictions': datasets.Value('int64'),
-                'references': datasets.Value('int64'),
+                'predictions': datasets.Value('string'),
+                'references': datasets.Value('string'),
             }),
             # Homepage of the module for documentation
             homepage="http://module.homepage",
@@ -90,6 +90,5 @@ class rouge_mn(evaluate.Metric):
         rouge = Rouge()
         scores = rouge.get_scores(predictions, references)[0]
         result = {key: scores[key]['f'] for key in scores.keys()}
-        """Returns the scores"""
-
+        
         return result

.ipynb_checkpoints/tests-checkpoint.py

@@ -0,0 +1,17 @@
+test_cases = [
+    {
+        "predictions": [0, 0],
+        "references": [1, 1],
+        "result": {"metric_score": 0}
+    },
+    {
+        "predictions": [1, 1],
+        "references": [1, 1],
+        "result": {"metric_score": 1}
+    },
+    {
+        "predictions": [1, 0],
+        "references": [1, 1],
+        "result": {"metric_score": 0.5}
+    }
+]

rouge_mn.py

@@ -71,8 +71,8 @@ class rouge_mn(evaluate.Metric):
             inputs_description=_KWARGS_DESCRIPTION,
             # This defines the format of each prediction and reference
             features=datasets.Features({
-                'predictions': datasets.Value('int64'),
-                'references': datasets.Value('int64'),
+                'predictions': datasets.Value('string'),
+                'references': datasets.Value('string'),
             }),
             # Homepage of the module for documentation
             homepage="http://module.homepage",
@@ -90,6 +90,5 @@ class rouge_mn(evaluate.Metric):
         rouge = Rouge()
         scores = rouge.get_scores(predictions, references)[0]
         result = {key: scores[key]['f'] for key in scores.keys()}
-        """Returns the scores"""
-
+        
         return result