add import

kaushiks_criteria.py

@@ -150,6 +150,8 @@ class kaushiks_criteria(evaluate.Metric):
         :param references:
         :return:
         """
+        from datasets import Dataset
+
         pred_ds = Dataset.from_dict({'text': predictions})
         refs_ds = Dataset.from_dict({'text': references})
         proc_ds = DatasetDict({'predictions': pred_ds, 'references': refs_ds})
@@ -173,6 +175,7 @@ class kaushiks_criteria(evaluate.Metric):
             predictions=proc_ds['predictions']['row_counts'],
             references=proc_ds['references']['row_counts']
         )['exact_match']
+
         return {
             'exact_match_headers': exact_match_headers,
             'exact_match_sep_row': exact_match_sep_row,