diff --git a/setup.py b/setup.py
index 0e8f508..f82bf3e 100644
--- a/setup.py
+++ b/setup.py
@@ -11,7 +11,7 @@
 # This call to setup() does all the work
 setup(
     name="torch_choice",
-    version="1.0.1",
+    version="1.0.2",
     description="A Pytorch Backend Library for Choice Modelling",
     long_description=README,
     long_description_content_type="text/markdown",
diff --git a/torch_choice/__init__.py b/torch_choice/__init__.py
index 80cafc5..01dd8a7 100644
--- a/torch_choice/__init__.py
+++ b/torch_choice/__init__.py
@@ -1,3 +1,3 @@
-__version__ = '1.0.1'
+__version__ = '1.0.2'
 
 from .utils.run_helper_lightning import run
\ No newline at end of file