Update info_utils.py

Author: ArjunJagdale

src/datasets/utils/info_utils.py

@@ -45,11 +45,12 @@ def verify_checksums(expected_checksums: Optional[dict], recorded_checksums: dic
     if expected_checksums is None:
         logger.info("Unable to verify checksums.")
         return
-    if len(set(expected_checksums) - set(recorded_checksums)) > 0:
-        raise ExpectedMoreDownloadedFilesError(str(set(expected_checksums) - set(recorded_checksums)))
-    if len(set(recorded_checksums) - set(expected_checksums)) > 0:
-        raise UnexpectedDownloadedFileError(str(set(recorded_checksums) - set(expected_checksums)))
-    bad_urls = [url for url in expected_checksums if expected_checksums[url] != recorded_checksums[url]]
+    bad_urls = [
+        url
+        for url in (set(recorded_checksums) & set(expected_checksums))
+        if expected_checksums[url] != recorded_checksums[url]
+    ]
+    
     for_verification_name = " for " + verification_name if verification_name is not None else ""
     if len(bad_urls) > 0:
         raise NonMatchingChecksumError(
@@ -64,13 +65,9 @@ def verify_splits(expected_splits: Optional[dict], recorded_splits: dict):
     if expected_splits is None:
         logger.info("Unable to verify splits sizes.")
         return
-    if len(set(expected_splits) - set(recorded_splits)) > 0:
-        raise ExpectedMoreSplitsError(str(set(expected_splits) - set(recorded_splits)))
-    if len(set(recorded_splits) - set(expected_splits)) > 0:
-        raise UnexpectedSplitsError(str(set(recorded_splits) - set(expected_splits)))
     bad_splits = [
         {"expected": expected_splits[name], "recorded": recorded_splits[name]}
-        for name in expected_splits
+        for name in (set(recorded_splits) & set(expected_splits))
         if expected_splits[name].num_examples != recorded_splits[name].num_examples
     ]
     if len(bad_splits) > 0: