[WIP] Support numpy 2.0

.circleci/config.yml

@@ -124,7 +124,7 @@ jobs:
 
           # we'll need packaging, numpy, and cython to build this. let install_requires do all the
           # rest of the work. We build with our lowest supported numpy version to make sure there is no regression
-          pip install "packaging>=17.1" "numpy~=1.19.3" "cython>=0.29,!=0.29.18,!=0.29.31"
+          pip install "packaging>=17.1" "numpy~=2.0.0" "cython~=3.0.0"
           python setup.py sdist
 
           cd dist
@@ -208,7 +208,7 @@ jobs:
     steps:
       - checkout
       - restore-whl-file:
-          key: "3.8"
+          key: "3.9"
       - deploy-to-pypi
 
   deploy-cpython39-whl:

.github/workflows/build_and_deploy.yml

@@ -53,7 +53,9 @@ jobs:
       # We build the source archive separately because of this: https://github.com/alkaline-ml/pmdarima/pull/136#discussion_r279781731
       # We build it first because of this: https://github.com/alkaline-ml/pmdarima/issues/448
       - name: Building source distribution
-        run: make version sdist
+        run: |
+          pip install -r build_tools/build_requirements.txt
+          make version sdist
 
       - name: Install cibuildwheel
         run: python -m pip install 'cibuildwheel<2.20.0'

.github/workflows/nightly_cron.yml

@@ -149,7 +149,7 @@ jobs:
       - name: Setting up Python
         uses: actions/setup-python@v4
         with:
-          python-version: '3.8'
+          python-version: '3.9'
 
       - name: Posting to Slack
         uses: 8398a7/action-slack@v3

Makefile

@@ -71,5 +71,5 @@ twine-check: bdist_wheel deploy-requirements
 	# Check that twine will parse the README acceptably
 	$(PYTHON) -m twine check dist/*
 
-version: requirements
+version:
 	@$(PYTHON) build_tools/get_tag.py

build_tools/build_requirements.txt

@@ -1,28 +1,13 @@
-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
+numpy==2.0.0
+scipy==1.13.0
+cython==3.0.0
 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
+setuptools>=38.6.0,!=50.0.0
+packaging>=17.1  # Bundled with setuptools, but want to be explicit

build_tools/circle/dind/install_and_test.sh

@@ -7,6 +7,8 @@ set -e
 # Make sure we're in the root PMDARIMA dir (mounted at /io)
 cd /io
 
+pip install -r /io/build_tools/build_requirements.txt
+
 make develop
 make testing-requirements
 make test-unit

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