save

kaushiks_criteria.py

@@ -150,7 +150,7 @@ class kaushiks_criteria(evaluate.Metric):
         :param references:
         :return:
         """
-        from datasets import Dataset
+        from datasets import Dataset, DatasetDict
 
         pred_ds = Dataset.from_dict({'text': predictions})
         refs_ds = Dataset.from_dict({'text': references})