Skip modin example for py37 (#47)

* Don't run example for py37 (which uses modin==0.12.1) Signed-off-by: Justin Yu <[email protected]> * Fix lint Signed-off-by: Justin Yu <[email protected]> --------- Signed-off-by: Justin Yu <[email protected]>
ray-project · Jun 14, 2023 · c7c894c · c7c894c
1 parent 394dc83
commit c7c894c
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/lightgbm_ray/examples/simple_modin.py b/lightgbm_ray/examples/simple_modin.py
@@ -3,6 +3,7 @@
 import numpy as np
 import pandas as pd
 import ray
+from packaging.version import Version
 from sklearn.utils import shuffle
 from xgboost_ray.data_sources.modin import MODIN_INSTALLED
 
@@ -17,6 +18,14 @@ def main(cpus_per_actor, num_actors):
         )
         return
 
+    import modin
+
+    if Version(modin.__version__) < Version("0.16.0") and Version(
+        ray.__version__
+    ) >= Version("2.6.0"):
+        print("modin<=0.16.0 is not compatible with ray>=2.6.0.")
+        return
+
     # Import modin after initializing Ray
     from modin.distributed.dataframe.pandas import from_partitions