Support numpy 2.0

build_tools/build_requirements.txt

@@ -1,28 +1,9 @@
-numpy==1.21.2; python_version < '3.10'
-numpy==1.21.6; python_version == '3.10' and platform_system != 'Windows'
-numpy==1.22.3; python_version == '3.10' and platform_system == 'Windows'
-numpy==1.23.2; python_version == '3.11'
-numpy==1.26.0; python_version == '3.12'
-scipy==1.3.2; python_version <= '3.8' and platform_machine != 'aarch64'
-scipy==1.5.3; python_version <= '3.8' and platform_machine == 'aarch64'
-scipy==1.5.4; python_version == '3.9'
-scipy==1.7.2; python_version == '3.10'
-scipy==1.9.3; python_version == '3.11'
-scipy==1.11.2; python_version == '3.12'
-statsmodels==0.13.2; python_version <= '3.10'
-statsmodels==0.13.3; python_version == '3.11'
-statsmodels==0.14.0; python_version == '3.12'
-cython>=0.29,!=0.29.18,!=0.29.31
-scikit-learn>=0.22
-pandas>=0.19
+# Other requirements needed in CI / deploying only
 patsy
 pytest
 pytest-mpl
 pytest-benchmark
-setuptools>=38.6.0,!=50.0.0
-packaging>=17.1  # Bundled with setuptools, but want to be explicit
 wheel
 twine>=1.13.0
 readme_renderer
 matplotlib
-urllib3

pmdarima/arima/tests/test_context.py

@@ -67,7 +67,7 @@ def test_nested_context():
 # Test a context honors the max duration
 def test_max_dur():
     # set arbitrarily low to guarantee will always pass after one iter
-    with StepwiseContext(max_dur=.5), \
+    with StepwiseContext(max_dur=.1), \
             pytest.warns(UserWarning) as uw:
 
         auto_arima(lynx, stepwise=True)
@@ -79,7 +79,7 @@ def test_max_dur():
 # Test that a context after the first will not inherit the first's attrs
 def test_subsequent_contexts():
     # Force a very fast fit
-    with StepwiseContext(max_dur=.5), \
+    with StepwiseContext(max_dur=.1), \
             pytest.warns(UserWarning):
         auto_arima(lynx, stepwise=True)
 

requirements.txt

@@ -1,10 +1,10 @@
-joblib>=0.11
-Cython>=0.29,!=0.29.18,!=0.29.31
-numpy>=1.21.2,<2.0.0
-pandas>=0.19
-scikit-learn>=0.22
-scipy>=1.3.2
-statsmodels>=0.13.2
-urllib3
+joblib>=1.2.0
+Cython>=3.0.6
+numpy>=2.0.0,<3
+pandas>=1.4,!=2.1.0
+scikit-learn>=1.5.2
+scipy>=1.10.0
+statsmodels>=0.14.4
+urllib3>=2.2.0
 setuptools>=38.6.0,!=50.0.0
 packaging>=17.1  # Bundled with setuptools, but want to be explicit