Update datastats.py

datastats.py

@@ -1,5 +1,6 @@
 """ DataStats metric. """
 
+import logging
 import functools
 from collections import Counter
 from multiprocessing import Pool
@@ -12,6 +13,8 @@ import datasets
 import evaluate
 from packaging import version
 
+logger = logging.getLogger(__name__)
+
 try:
     _en = spacy.load('en_core_web_sm')
 except OSError as stderr:
@@ -299,7 +302,7 @@ class DataStats(evaluate.Metric):
         coverage = float(results['coverage'])
         density = float(results['density'])
         compression = float(results['compression'])
-        print(coverage, density, compression)
+        logger.info(coverage, density, compression)
         return {
             "coverage": coverage, 
             "density": density,