Update datastats.py

datastats.py

@@ -271,8 +271,8 @@ class DataStats(evaluate.Metric):
             features=[
                 datasets.Features(
                     {
-                        "summaries": datasets.Value("string", id="sequence"),
-                        "documents": datasets.Value("string", id="sequence"),
+                        "predictions": datasets.Value("string", id="sequence"),
+                        "references": datasets.Value("string", id="sequence"),
                     }
                 ),
             ],
@@ -285,8 +285,8 @@ class DataStats(evaluate.Metric):
 
     def _compute(
         self,
-        summaries, 
-        documents, 
+        predictions, 
+        references, 
         n_gram: int = 3, 
         n_workers: int = 4, 
         lowercase: bool = False, 
@@ -296,7 +296,7 @@ class DataStats(evaluate.Metric):
         # logger.info(predictions)
         # logger.info(references)
         datastats = DataStatsMetric(n_gram, n_workers, lowercase, tokenize)
-        results = datastats.evaluate_batch(summaries, documents)
+        results = datastats.evaluate_batch(predictions, references)
         coverage = float(results['coverage'])
         density = float(results['density'])
         compression = float(results['compression'])