Add checksums path to tid2013 builder

Devparihar5 · Oct 6, 2019 · 607044e · 607044e
1 parent 48791b8
commit 607044e
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/imquality/__init__.py b/imquality/__init__.py
@@ -1 +1 @@
-__version__ = "1.2.1"
+__version__ = "1.2.2"
diff --git a/imquality/datasets/tid2013.py b/imquality/datasets/tid2013.py
@@ -3,6 +3,10 @@
 import tensorflow as tf
 import tensorflow_datasets.public_api as tfds
 
+from . import CHECKSUMS_PATH
+
+tfds.download.add_checksums_dir(CHECKSUMS_PATH)
+
 CITATION = r"""
 @article{ponomarenko2015image,
   title={Image database TID2013: Peculiarities, results and perspectives},