diff --git a/lightgbm_ray/__init__.py b/lightgbm_ray/__init__.py
index 3392b1e..35a5277 100644
--- a/lightgbm_ray/__init__.py
+++ b/lightgbm_ray/__init__.py
@@ -5,7 +5,7 @@
 
 from lightgbm_ray.sklearn import RayLGBMClassifier, RayLGBMRegressor
 
-__version__ = "0.1.6"
+__version__ = "0.1.7"
 
 __all__ = [
     "__version__", "RayParams", "RayDMatrix", "RayDeviceQuantileDMatrix",
diff --git a/setup.py b/setup.py
index 7a23523..feab3fb 100644
--- a/setup.py
+++ b/setup.py
@@ -3,7 +3,7 @@
 setup(
     name="lightgbm_ray",
     packages=find_packages(where=".", include="lightgbm_ray*"),
-    version="0.1.6",
+    version="0.1.7",
     author="Ray Team",
     description="A Ray backend for distributed LightGBM",
     license="Apache 2.0",