diff --git a/setup.cfg b/setup.cfg
index f0dd4d8..f152300 100644
--- a/setup.cfg
+++ b/setup.cfg
@@ -41,7 +41,6 @@ install_requires =
     tensorflow
     torch==1.10.0
     torchvision==0.11.1
-    uq360==0.2
     wandb
     xlrd==2.0.1
 
@@ -56,6 +55,7 @@ benchmarks =
     aix360
     alibi-detect==0.9.1
     tensorflow>2
+    uq360==0.2
 
 contribute =
     black
diff --git a/src/data_suite/version.py b/src/data_suite/version.py
new file mode 100644
index 0000000..e835c08
--- /dev/null
+++ b/src/data_suite/version.py
@@ -0,0 +1,4 @@
+__version__ = "0.0.2"
+
+MAJOR_VERSION = ".".join(__version__.split(".")[:-1])
+PATCH_VERSION = __version__.split(".")[-1]