Allow small rounding errors for bee_waggle_6

For this dataset we compute two of the features, and small rounding
errors can occur on different systems. This fix allows these
errors at least on Travis.

1 files changed, 8 insertions, 2 deletions

diff --git a/utils/check_checksums.py b/utils/check_checksums.py
index 01dcd99..0d66400 100644
--- a/utils/check_checksums.py
+++ b/utils/check_checksums.py
@@ -73,8 +73,14 @@ def main():
         if not fname in data_files:
             raise FileNotFoundError("Missing data file: %s" % fname)
         md5 = md5sum(data_files[fname])
-        if not md5 == checksums[fname]:
-            raise ValueError(
+        if isinstance(checksums[fname], list):
+            if not md5 in checksums[fname]:
+                raise ValueError(
+                    "Checksums don't match for file: %s" % (data_files[fname])
+                    )
+        else:
+            if not md5 == checksums[fname]:
+                raise ValueError(
                 "Checksums don't match for file: %s" % (data_files[fname])
             )