diff --git a/modin/numpy/__init__.py b/modin/numpy/__init__.py index 2c0558669e0..4cf4f23f984 100644 --- 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 index 61aaef387d9..5f103f66d77 100644 --- 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",