From 31122a3b83d842fb41dd089c18dd2a4c9cbe6661 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Jos=C3=A9=20Morales?= <jmoralz92@gmail.com>
Date: Thu, 12 Sep 2024 12:21:52 -0600
Subject: [PATCH 01/13] ci: build wheels and run tests on same job

---
 .github/workflows/ci.yaml | 115 ++++++++------------------------------
 settings.ini              |   2 +-
 2 files changed, 25 insertions(+), 92 deletions(-)

diff --git a/.github/workflows/ci.yaml b/.github/workflows/ci.yaml
index 1b93d6253..86286cbe5 100644
--- a/.github/workflows/ci.yaml
+++ b/.github/workflows/ci.yaml
@@ -14,9 +14,10 @@ env:
   NIXTLA_NUMBA_CACHE: "1"
 
 jobs:
-  build_wheels:
-    name: Build wheels for cp${{ matrix.python-version }}-${{ matrix.os-platform[1] }}
+  run-tests:
+    name: Run tests for cp${{ matrix.python-version }}-${{ matrix.os-platform[1] }}
     runs-on: ${{ matrix.os-platform[0] }}
+    timeout-minutes: 60
     strategy:
       fail-fast: false
       matrix:
@@ -40,12 +41,29 @@ jobs:
         with:
           platforms: arm64
 
-      - name: Build wheels
+      - name: Build wheels and run all tests
+        if: matrix.os-platform[0] == 'ubuntu-latest'
+        uses: pypa/cibuildwheel@bd033a44476646b606efccdd5eed92d5ea1d77ad # v2.20.0
+        env:
+          CIBW_BUILD: cp${{ matrix.python-version }}-${{ matrix.os-platform[1] }}
+          CIBW_TEST_COMMAND: >
+            pip install uv &&
+            uv pip install --system -r setup.py --extra all &&
+            nbdev_test --skip_file_re '(distributed|prophet).*.ipynb' --pause 1.0 --do_print --timing &&
+            pytest --durations=0 action_files
+
+      - name: Build wheels and run local tests
+        if: matrix.os-platform[0] != 'ubuntu-latest'
         uses: pypa/cibuildwheel@bd033a44476646b606efccdd5eed92d5ea1d77ad # v2.20.0
         env:
           CIBW_BUILD: cp${{ matrix.python-version }}-${{ matrix.os-platform[1] }}
+          CIBW_TEST_COMMAND: >
+            pip install uv &&
+            uv pip install --system -r setup.py --extra dev --extra plotly
+            nbdev_test --skip_file_re '(distributed|prophet).*.ipynb' --pause 1.0 --do_print --timing
 
       - uses: actions/upload-artifact@50769540e7f4bd5e21e526ee35c689e35e0d6874 # v4.4.0
+        if: github.event_name == 'push'
         with:
           name: artifact-cp${{ matrix.python-version }}-${{ matrix.os-platform[1] }}
           path: wheelhouse/*.whl
@@ -53,7 +71,7 @@ jobs:
 
   publish-nightly-wheels:
     if: github.event_name == 'push'
-    needs: [build_wheels]
+    needs: [run-tests]
     env:
       AWS_ACCESS_KEY_ID: ${{ secrets.AWS_ACCESS_KEY_ID_NIXTLA_PACKAGES }}
       AWS_SECRET_ACCESS_KEY: ${{ secrets.AWS_SECRET_ACCESS_KEY_NIXTLA_PACKAGES }}
@@ -79,106 +97,21 @@ jobs:
           python action_files/update_index.py
           aws s3 sync dist s3://nixtla-packages/statsforecast/
 
-  run-local-tests:
-    needs: [build_wheels]
-    runs-on: ${{ matrix.os }}
-    timeout-minutes: 60
-    strategy:
-      fail-fast: false
-      matrix:
-        os: [macos-13, macos-14, windows-latest]
-        python-version: ["3.8", "3.9", "3.10", "3.11", "3.12"]
-    steps:
-      - name: Clone repo
-        uses: actions/checkout@692973e3d937129bcbf40652eb9f2f61becf3332 # v4.1.7
-        with:
-          submodules: "true"
-
-      - name: Set up environment
-        uses: actions/setup-python@f677139bbe7f9c59b41e40162b753c062f5d49a3 # v5.2.0
-        with:
-          python-version: ${{ matrix.python-version }}
-
-      - name: Install pip requirements
-        run: pip install uv && uv pip install --system -r setup.py --extra dev --extra plotly
-
-      - name: Download wheels
-        uses: actions/download-artifact@fa0a91b85d4f404e444e00e005971372dc801d16 # v4.1.8
-        with:
-          path: dist
-          pattern: "artifact-*"
-          merge-multiple: true
-
-      - name: Install the library
-        run: pip install --no-index --find-links dist statsforecast
-
-      - name: Run local tests
-        run: nbdev_test --skip_file_re '(distributed|prophet).*.ipynb' --pause 1.0 --do_print --timing
-
-  run-tests:
-    needs: [build_wheels]
+  test-m3-performance:
     runs-on: ubuntu-latest
-    timeout-minutes: 60
-    strategy:
-      fail-fast: false
-      matrix:
-        python-version: ["3.8", "3.9", "3.10", "3.11", "3.12"]
     steps:
       - name: Clone repo
         uses: actions/checkout@692973e3d937129bcbf40652eb9f2f61becf3332 # v4.1.7
         with:
           submodules: "true"
 
-      - name: Set up environment
-        uses: actions/setup-python@f677139bbe7f9c59b41e40162b753c062f5d49a3 # v5.2.0
-        with:
-          python-version: ${{ matrix.python-version }}
-
-      - name: Install pip requirements
-        run: pip install uv && uv pip install --system -r setup.py --extra all
-
-      - name: Download wheels
-        uses: actions/download-artifact@fa0a91b85d4f404e444e00e005971372dc801d16 # v4.1.8
-        with:
-          path: dist
-          pattern: "artifact-*"
-          merge-multiple: true
-
-      - name: Install the library
-        run: pip install --no-index --find-links dist statsforecast
-
-      - name: Run tests
-        run: nbdev_test --skip_file_re '(distributed|prophet).*.ipynb' --pause 1.0 --do_print --timing
-
-      - name: Run integration tests
-        run: |
-          uv pip install --system pytest
-          pytest --durations=0 action_files
-
-  test-m3-performance:
-    needs: [build_wheels]
-    runs-on: ubuntu-latest
-    steps:
-      - name: Clone repo
-        uses: actions/checkout@692973e3d937129bcbf40652eb9f2f61becf3332 # v4.1.7
-
       - name: Set up environment
         uses: actions/setup-python@f677139bbe7f9c59b41e40162b753c062f5d49a3 # v5.2.0
         with:
           python-version: "3.10"
 
-      - name: Install pip requirements
-        run: pip install uv && uv pip install --system -r setup.py --extra dev
-
-      - name: Download wheels
-        uses: actions/download-artifact@fa0a91b85d4f404e444e00e005971372dc801d16 # v4.1.8
-        with:
-          path: dist
-          pattern: "artifact-*"
-          merge-multiple: true
-
       - name: Install the library
-        run: pip install --no-index --find-links dist statsforecast
+        run: pip install uv && uv pip install --system ".[dev]"
 
       - name: Run M3 experiment
         run: |
diff --git a/settings.ini b/settings.ini
index 83ac737ac..fcdd51151 100644
--- a/settings.ini
+++ b/settings.ini
@@ -21,7 +21,7 @@ ray_requirements = fugue[ray]>=0.8.1 protobuf>=3.15.3,<4.0.0 numpy<2 pandas<2.2
 dask_requirements = fugue[dask]>=0.8.1
 spark_requirements = fugue[spark]>=0.8.1
 plotly_requirements = plotly plotly-resampler
-dev_requirements = black datasetsforecast fire nbdev nbformat nbdev_plotly pandas[plot] pmdarima polars[numpy] pre-commit prophet pyarrow pybind11 scikit-learn setuptools<70 supersmoother
+dev_requirements = black datasetsforecast fire nbdev nbformat nbdev_plotly pandas[plot] pmdarima polars[numpy] pre-commit prophet pyarrow pybind11 pytest scikit-learn setuptools<70 supersmoother
 nbs_path = nbs
 doc_path = _docs
 recursive = True

From afa0f810f9e097b23cbeb76dd95c494f26ba1c9d Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Jos=C3=A9=20Morales?= <jmoralz92@gmail.com>
Date: Thu, 12 Sep 2024 12:33:20 -0600
Subject: [PATCH 02/13] use project placeholder

---
 .github/workflows/ci.yaml | 11 ++++++-----
 1 file changed, 6 insertions(+), 5 deletions(-)

diff --git a/.github/workflows/ci.yaml b/.github/workflows/ci.yaml
index 86286cbe5..1c110c0a5 100644
--- a/.github/workflows/ci.yaml
+++ b/.github/workflows/ci.yaml
@@ -46,11 +46,12 @@ jobs:
         uses: pypa/cibuildwheel@bd033a44476646b606efccdd5eed92d5ea1d77ad # v2.20.0
         env:
           CIBW_BUILD: cp${{ matrix.python-version }}-${{ matrix.os-platform[1] }}
+          CIBW_TEST_SKIP: "*linux_aarch64"
           CIBW_TEST_COMMAND: >
             pip install uv &&
-            uv pip install --system -r setup.py --extra all &&
-            nbdev_test --skip_file_re '(distributed|prophet).*.ipynb' --pause 1.0 --do_print --timing &&
-            pytest --durations=0 action_files
+            uv pip install --system -r {project}/setup.py --extra all &&
+            nbdev_test --path {project}/nbs/ --skip_file_re '(distributed|prophet).*.ipynb' --pause 1.0 --do_print --timing &&
+            pytest --durations=0 {project}/action_files
 
       - name: Build wheels and run local tests
         if: matrix.os-platform[0] != 'ubuntu-latest'
@@ -59,8 +60,8 @@ jobs:
           CIBW_BUILD: cp${{ matrix.python-version }}-${{ matrix.os-platform[1] }}
           CIBW_TEST_COMMAND: >
             pip install uv &&
-            uv pip install --system -r setup.py --extra dev --extra plotly
-            nbdev_test --skip_file_re '(distributed|prophet).*.ipynb' --pause 1.0 --do_print --timing
+            uv pip install --system -r {project}/setup.py --extra dev --extra plotly &&
+            nbdev_test --path {project}/nbs/ --skip_file_re '(distributed|prophet).*.ipynb' --pause 1.0 --do_print --timing
 
       - uses: actions/upload-artifact@50769540e7f4bd5e21e526ee35c689e35e0d6874 # v4.4.0
         if: github.event_name == 'push'

From 31e0a3da4d24346a63a30716f09bd2fdac75fd35 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Jos=C3=A9=20Morales?= <jmoralz92@gmail.com>
Date: Thu, 12 Sep 2024 12:43:16 -0600
Subject: [PATCH 03/13] setup uv

---
 .github/workflows/ci.yaml | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/.github/workflows/ci.yaml b/.github/workflows/ci.yaml
index 1c110c0a5..e363eb36b 100644
--- a/.github/workflows/ci.yaml
+++ b/.github/workflows/ci.yaml
@@ -18,6 +18,8 @@ jobs:
     name: Run tests for cp${{ matrix.python-version }}-${{ matrix.os-platform[1] }}
     runs-on: ${{ matrix.os-platform[0] }}
     timeout-minutes: 60
+    env:
+      CIBW_BUILD_FRONTEND: "build[uv]"
     strategy:
       fail-fast: false
       matrix:
@@ -41,6 +43,11 @@ jobs:
         with:
           platforms: arm64
 
+      - name: Set up uv
+        uses: astral-sh/setup-uv@d70817ce85e489f62dba96dc5d8f7a926e48bdd7 # v2.1.1
+        with:
+          version: "latest"
+
       - name: Build wheels and run all tests
         if: matrix.os-platform[0] == 'ubuntu-latest'
         uses: pypa/cibuildwheel@bd033a44476646b606efccdd5eed92d5ea1d77ad # v2.20.0
@@ -48,7 +55,6 @@ jobs:
           CIBW_BUILD: cp${{ matrix.python-version }}-${{ matrix.os-platform[1] }}
           CIBW_TEST_SKIP: "*linux_aarch64"
           CIBW_TEST_COMMAND: >
-            pip install uv &&
             uv pip install --system -r {project}/setup.py --extra all &&
             nbdev_test --path {project}/nbs/ --skip_file_re '(distributed|prophet).*.ipynb' --pause 1.0 --do_print --timing &&
             pytest --durations=0 {project}/action_files
@@ -59,7 +65,6 @@ jobs:
         env:
           CIBW_BUILD: cp${{ matrix.python-version }}-${{ matrix.os-platform[1] }}
           CIBW_TEST_COMMAND: >
-            pip install uv &&
             uv pip install --system -r {project}/setup.py --extra dev --extra plotly &&
             nbdev_test --path {project}/nbs/ --skip_file_re '(distributed|prophet).*.ipynb' --pause 1.0 --do_print --timing
 

From 35fc349ffe79b5af08c532ec843814450a621f9a Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Jos=C3=A9=20Morales?= <jmoralz92@gmail.com>
Date: Thu, 12 Sep 2024 12:50:58 -0600
Subject: [PATCH 04/13] install into venv and remove skip setting

---
 .github/workflows/ci.yaml | 4 ++--
 pyproject.toml            | 1 -
 2 files changed, 2 insertions(+), 3 deletions(-)

diff --git a/.github/workflows/ci.yaml b/.github/workflows/ci.yaml
index e363eb36b..c5a0e4110 100644
--- a/.github/workflows/ci.yaml
+++ b/.github/workflows/ci.yaml
@@ -55,7 +55,7 @@ jobs:
           CIBW_BUILD: cp${{ matrix.python-version }}-${{ matrix.os-platform[1] }}
           CIBW_TEST_SKIP: "*linux_aarch64"
           CIBW_TEST_COMMAND: >
-            uv pip install --system -r {project}/setup.py --extra all &&
+            uv pip install -r {project}/setup.py --extra all &&
             nbdev_test --path {project}/nbs/ --skip_file_re '(distributed|prophet).*.ipynb' --pause 1.0 --do_print --timing &&
             pytest --durations=0 {project}/action_files
 
@@ -65,7 +65,7 @@ jobs:
         env:
           CIBW_BUILD: cp${{ matrix.python-version }}-${{ matrix.os-platform[1] }}
           CIBW_TEST_COMMAND: >
-            uv pip install --system -r {project}/setup.py --extra dev --extra plotly &&
+            uv pip install -r {project}/setup.py --extra dev --extra plotly &&
             nbdev_test --path {project}/nbs/ --skip_file_re '(distributed|prophet).*.ipynb' --pause 1.0 --do_print --timing
 
       - uses: actions/upload-artifact@50769540e7f4bd5e21e526ee35c689e35e0d6874 # v4.4.0
diff --git a/pyproject.toml b/pyproject.toml
index 375c37033..6df73f1b3 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -5,7 +5,6 @@ build-backend = "setuptools.build_meta"
 [tool.cibuildwheel]
 archs = "all"
 build-verbosity = 1
-test-skip = "*"
 
 [tool.ruff]
 target-version = "py38"

From d124e7d91063f05fa162bbb67a7b943f56b31cfb Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Jos=C3=A9=20Morales?= <jmoralz92@gmail.com>
Date: Thu, 12 Sep 2024 12:56:24 -0600
Subject: [PATCH 05/13] try double quotes

---
 .github/workflows/ci.yaml | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/.github/workflows/ci.yaml b/.github/workflows/ci.yaml
index c5a0e4110..1ba553fe4 100644
--- a/.github/workflows/ci.yaml
+++ b/.github/workflows/ci.yaml
@@ -56,7 +56,7 @@ jobs:
           CIBW_TEST_SKIP: "*linux_aarch64"
           CIBW_TEST_COMMAND: >
             uv pip install -r {project}/setup.py --extra all &&
-            nbdev_test --path {project}/nbs/ --skip_file_re '(distributed|prophet).*.ipynb' --pause 1.0 --do_print --timing &&
+            nbdev_test --path {project}/nbs/ --skip_file_re "(distributed|prophet).*.ipynb" --pause 1.0 --do_print --timing &&
             pytest --durations=0 {project}/action_files
 
       - name: Build wheels and run local tests
@@ -66,7 +66,7 @@ jobs:
           CIBW_BUILD: cp${{ matrix.python-version }}-${{ matrix.os-platform[1] }}
           CIBW_TEST_COMMAND: >
             uv pip install -r {project}/setup.py --extra dev --extra plotly &&
-            nbdev_test --path {project}/nbs/ --skip_file_re '(distributed|prophet).*.ipynb' --pause 1.0 --do_print --timing
+            nbdev_test --path {project}/nbs/ --skip_file_re "(distributed|prophet).*.ipynb" --pause 1.0 --do_print --timing
 
       - uses: actions/upload-artifact@50769540e7f4bd5e21e526ee35c689e35e0d6874 # v4.4.0
         if: github.event_name == 'push'

From b6082ef2c0e99b3679fbb5805a4ef672894914cb Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Jos=C3=A9=20Morales?= <jmoralz92@gmail.com>
Date: Thu, 12 Sep 2024 13:26:04 -0600
Subject: [PATCH 06/13] install java?

---
 .github/workflows/ci.yaml | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/.github/workflows/ci.yaml b/.github/workflows/ci.yaml
index 1ba553fe4..1fa5b0c6b 100644
--- a/.github/workflows/ci.yaml
+++ b/.github/workflows/ci.yaml
@@ -54,13 +54,14 @@ jobs:
         env:
           CIBW_BUILD: cp${{ matrix.python-version }}-${{ matrix.os-platform[1] }}
           CIBW_TEST_SKIP: "*linux_aarch64"
+          CIBW_BEFORE_TEST: sudo apt install default-jre
           CIBW_TEST_COMMAND: >
             uv pip install -r {project}/setup.py --extra all &&
-            nbdev_test --path {project}/nbs/ --skip_file_re "(distributed|prophet).*.ipynb" --pause 1.0 --do_print --timing &&
             pytest --durations=0 {project}/action_files
 
       - name: Build wheels and run local tests
-        if: matrix.os-platform[0] != 'ubuntu-latest'
+        # if: matrix.os-platform[0] != 'ubuntu-latest'
+        if: false
         uses: pypa/cibuildwheel@bd033a44476646b606efccdd5eed92d5ea1d77ad # v2.20.0
         env:
           CIBW_BUILD: cp${{ matrix.python-version }}-${{ matrix.os-platform[1] }}

From e4c90f61f121aa754c33ed9917e7687c204ec6fa Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Jos=C3=A9=20Morales?= <jmoralz92@gmail.com>
Date: Thu, 12 Sep 2024 13:28:17 -0600
Subject: [PATCH 07/13] without sudo?

---
 .github/workflows/ci.yaml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/.github/workflows/ci.yaml b/.github/workflows/ci.yaml
index 1fa5b0c6b..805f5315d 100644
--- a/.github/workflows/ci.yaml
+++ b/.github/workflows/ci.yaml
@@ -54,7 +54,7 @@ jobs:
         env:
           CIBW_BUILD: cp${{ matrix.python-version }}-${{ matrix.os-platform[1] }}
           CIBW_TEST_SKIP: "*linux_aarch64"
-          CIBW_BEFORE_TEST: sudo apt install default-jre
+          CIBW_BEFORE_TEST: apt install default-jre
           CIBW_TEST_COMMAND: >
             uv pip install -r {project}/setup.py --extra all &&
             pytest --durations=0 {project}/action_files

From a9350a9e60fdc66796246e416b7ce8cff7059b50 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Jos=C3=A9=20Morales?= <jmoralz92@gmail.com>
Date: Thu, 12 Sep 2024 13:35:57 -0600
Subject: [PATCH 08/13] maybe?

---
 .github/workflows/ci.yaml | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/.github/workflows/ci.yaml b/.github/workflows/ci.yaml
index 805f5315d..35b4df75c 100644
--- a/.github/workflows/ci.yaml
+++ b/.github/workflows/ci.yaml
@@ -43,6 +43,13 @@ jobs:
         with:
           platforms: arm64
 
+      - name: Set up Java
+        if: matrix.os-platform[1] == 'manylinux_x86_64'
+        uses: actions/setup-java@2dfa2011c5b2a0f1489bf9e433881c92c1631f88 # v4.3.0
+        with:
+          distribution: "temurin"
+          java-version: "21"
+
       - name: Set up uv
         uses: astral-sh/setup-uv@d70817ce85e489f62dba96dc5d8f7a926e48bdd7 # v2.1.1
         with:
@@ -54,7 +61,6 @@ jobs:
         env:
           CIBW_BUILD: cp${{ matrix.python-version }}-${{ matrix.os-platform[1] }}
           CIBW_TEST_SKIP: "*linux_aarch64"
-          CIBW_BEFORE_TEST: apt install default-jre
           CIBW_TEST_COMMAND: >
             uv pip install -r {project}/setup.py --extra all &&
             pytest --durations=0 {project}/action_files

From 854b1a00e5e6b3afa0b8d5d2f6e3d027e968e308 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Jos=C3=A9=20Morales?= <jmoralz92@gmail.com>
Date: Thu, 12 Sep 2024 14:00:33 -0600
Subject: [PATCH 09/13] container uses centos

---
 .github/workflows/ci.yaml | 8 +-------
 1 file changed, 1 insertion(+), 7 deletions(-)

diff --git a/.github/workflows/ci.yaml b/.github/workflows/ci.yaml
index 35b4df75c..2f555d8aa 100644
--- a/.github/workflows/ci.yaml
+++ b/.github/workflows/ci.yaml
@@ -43,13 +43,6 @@ jobs:
         with:
           platforms: arm64
 
-      - name: Set up Java
-        if: matrix.os-platform[1] == 'manylinux_x86_64'
-        uses: actions/setup-java@2dfa2011c5b2a0f1489bf9e433881c92c1631f88 # v4.3.0
-        with:
-          distribution: "temurin"
-          java-version: "21"
-
       - name: Set up uv
         uses: astral-sh/setup-uv@d70817ce85e489f62dba96dc5d8f7a926e48bdd7 # v2.1.1
         with:
@@ -71,6 +64,7 @@ jobs:
         uses: pypa/cibuildwheel@bd033a44476646b606efccdd5eed92d5ea1d77ad # v2.20.0
         env:
           CIBW_BUILD: cp${{ matrix.python-version }}-${{ matrix.os-platform[1] }}
+          CIBW_BEFORE_TEST: yum install -y java
           CIBW_TEST_COMMAND: >
             uv pip install -r {project}/setup.py --extra dev --extra plotly &&
             nbdev_test --path {project}/nbs/ --skip_file_re "(distributed|prophet).*.ipynb" --pause 1.0 --do_print --timing

From 92cc6b40adc302f81ec60209ac70d20f00ad43c8 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Jos=C3=A9=20Morales?= <jmoralz92@gmail.com>
Date: Thu, 12 Sep 2024 14:00:56 -0600
Subject: [PATCH 10/13] test spark directly

---
 .github/workflows/ci.yaml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/.github/workflows/ci.yaml b/.github/workflows/ci.yaml
index 2f555d8aa..5da252453 100644
--- a/.github/workflows/ci.yaml
+++ b/.github/workflows/ci.yaml
@@ -56,7 +56,7 @@ jobs:
           CIBW_TEST_SKIP: "*linux_aarch64"
           CIBW_TEST_COMMAND: >
             uv pip install -r {project}/setup.py --extra all &&
-            pytest --durations=0 {project}/action_files
+            pytest --durations=0 {project}/action_files/test_spark.py
 
       - name: Build wheels and run local tests
         # if: matrix.os-platform[0] != 'ubuntu-latest'

From 6393b88d32e4cbb93bcca78a821d94fbfc8652d6 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Jos=C3=A9=20Morales?= <jmoralz92@gmail.com>
Date: Thu, 12 Sep 2024 14:04:03 -0600
Subject: [PATCH 11/13] set in correct place

---
 .github/workflows/ci.yaml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/.github/workflows/ci.yaml b/.github/workflows/ci.yaml
index 5da252453..575f813f9 100644
--- a/.github/workflows/ci.yaml
+++ b/.github/workflows/ci.yaml
@@ -54,6 +54,7 @@ jobs:
         env:
           CIBW_BUILD: cp${{ matrix.python-version }}-${{ matrix.os-platform[1] }}
           CIBW_TEST_SKIP: "*linux_aarch64"
+          CIBW_BEFORE_TEST: yum install -y java
           CIBW_TEST_COMMAND: >
             uv pip install -r {project}/setup.py --extra all &&
             pytest --durations=0 {project}/action_files/test_spark.py
@@ -64,7 +65,6 @@ jobs:
         uses: pypa/cibuildwheel@bd033a44476646b606efccdd5eed92d5ea1d77ad # v2.20.0
         env:
           CIBW_BUILD: cp${{ matrix.python-version }}-${{ matrix.os-platform[1] }}
-          CIBW_BEFORE_TEST: yum install -y java
           CIBW_TEST_COMMAND: >
             uv pip install -r {project}/setup.py --extra dev --extra plotly &&
             nbdev_test --path {project}/nbs/ --skip_file_re "(distributed|prophet).*.ipynb" --pause 1.0 --do_print --timing

From ae8ecd70a33a2f146de01024f8532c2f7c6b26a7 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Jos=C3=A9=20Morales?= <jmoralz92@gmail.com>
Date: Thu, 12 Sep 2024 14:09:00 -0600
Subject: [PATCH 12/13] restore tests

---
 .github/workflows/ci.yaml | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/.github/workflows/ci.yaml b/.github/workflows/ci.yaml
index 575f813f9..51480fd9a 100644
--- a/.github/workflows/ci.yaml
+++ b/.github/workflows/ci.yaml
@@ -57,11 +57,11 @@ jobs:
           CIBW_BEFORE_TEST: yum install -y java
           CIBW_TEST_COMMAND: >
             uv pip install -r {project}/setup.py --extra all &&
-            pytest --durations=0 {project}/action_files/test_spark.py
+            nbdev_test --path {project}/nbs/ --skip_file_re "(distributed|prophet).*.ipynb" --pause 1.0 --do_print --timing &&
+            pytest --durations=0 {project}/action_files
 
       - name: Build wheels and run local tests
-        # if: matrix.os-platform[0] != 'ubuntu-latest'
-        if: false
+        if: matrix.os-platform[0] != 'ubuntu-latest'
         uses: pypa/cibuildwheel@bd033a44476646b606efccdd5eed92d5ea1d77ad # v2.20.0
         env:
           CIBW_BUILD: cp${{ matrix.python-version }}-${{ matrix.os-platform[1] }}

From 796252e3f4f94ec3135f02847ef1b5a5c67ade21 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Jos=C3=A9=20Morales?= <jmoralz92@gmail.com>
Date: Thu, 12 Sep 2024 14:33:41 -0600
Subject: [PATCH 13/13] try setting cache env var

---
 .github/workflows/ci.yaml | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/.github/workflows/ci.yaml b/.github/workflows/ci.yaml
index 51480fd9a..c7654a3c5 100644
--- a/.github/workflows/ci.yaml
+++ b/.github/workflows/ci.yaml
@@ -19,7 +19,8 @@ jobs:
     runs-on: ${{ matrix.os-platform[0] }}
     timeout-minutes: 60
     env:
-      CIBW_BUILD_FRONTEND: "build[uv]"
+      CIBW_BUILD_FRONTEND: build[uv]
+      CIBW_ENVIRONMENT: NIXTLA_NUMBA_CACHE=1
     strategy:
       fail-fast: false
       matrix: