diff --git a/requirements.txt b/requirements.txt
index 3ddb2ac..9a13add 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -4,3 +4,4 @@ pandas
 torch
 tensorboard
 packaging
+datasets
diff --git a/version.py b/version.py
index fa9c4ec..2792152 100644
--- a/version.py
+++ b/version.py
@@ -1 +1 @@
-__version__ = '0.0.6'
+__version__ = '0.0.7'