fix numpy version

Signed-off-by: Anatoly Myachev <anatoly.myachev@intel.com>
anmyachev · Jun 13, 2024 · 7dc0a08 · 7dc0a08
1 parent db61e17
commit 7dc0a08
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/modin/numpy/__init__.py b/modin/numpy/__init__.py
@@ -20,7 +20,7 @@
 from .array_shaping import append, hstack, ravel, shape, split, transpose
 from .constants import e, euler_gamma, inf, nan, newaxis, pi
 
-if version.parse(numpy.__version__) < version.parse("2.0.0"):
+if version.parse(numpy.__version__) < version.parse("2.0.0b1"):
     from .constants import (
         NAN,
         NINF,
@@ -169,7 +169,7 @@ def where(condition, x=None, y=None):
     "append",
     "tri",
 ]
-if version.parse(numpy.__version__) < version.parse("2.0.0"):
+if version.parse(numpy.__version__) < version.parse("2.0.0b1"):
     __all__ += [
         "Inf",
         "Infinity",

diff --git a/modin/numpy/constants.py b/modin/numpy/constants.py
@@ -15,7 +15,7 @@
 from numpy import e, euler_gamma, inf, nan, newaxis, pi
 from packaging import version
 
-if version.parse(numpy.__version__) < version.parse("2.0.0"):
+if version.parse(numpy.__version__) < version.parse("2.0.0b1"):
     from numpy import NAN, NINF, NZERO, PINF, PZERO, Inf, Infinity, NaN, infty
 
 __all__ = [
@@ -27,7 +27,7 @@
     "pi",
 ]
 
-if version.parse(numpy.__version__) < version.parse("2.0.0"):
+if version.parse(numpy.__version__) < version.parse("2.0.0b1"):
     __all__ += [
         "Inf",
         "Infinity",