Skip to content

Commit e2879cf

Browse files
ENH: GPU Rocket 2.0: CuPy Backend, Exact CPU Parity, 100x Speedup
1 parent e497333 commit e2879cf

15 files changed

Lines changed: 672 additions & 382 deletions

File tree

.github/workflows/fast_release.yml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,7 @@ jobs:
2222
python -m build
2323
2424
- name: Store build files
25-
uses: actions/upload-artifact@v6
25+
uses: actions/upload-artifact@v5
2626
with:
2727
name: dist
2828
path: dist/*
@@ -38,7 +38,7 @@ jobs:
3838
id-token: write
3939

4040
steps:
41-
- uses: actions/download-artifact@v7
41+
- uses: actions/download-artifact@v6
4242
with:
4343
name: dist
4444
path: dist

.github/workflows/periodic_tests.yml

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -85,7 +85,7 @@ jobs:
8585
shell: bash
8686

8787
- name: Save new cache
88-
uses: actions/cache/save@v5
88+
uses: actions/cache/save@v4
8989
with:
9090
path: ${{ github.workspace }}/.numba_cache
9191
# Save cache with the current date (ENV set in numba_cache action)
@@ -150,7 +150,7 @@ jobs:
150150
run: python -m pytest -n logical
151151

152152
- name: Save new cache
153-
uses: actions/cache/save@v5
153+
uses: actions/cache/save@v4
154154
with:
155155
path: ${{ github.workspace }}/.numba_cache
156156
# Save cache with the current date (ENV set in numba_cache action)
@@ -209,7 +209,7 @@ jobs:
209209
run: python -m pytest -n logical
210210

211211
- name: Save new cache
212-
uses: actions/cache/save@v5
212+
uses: actions/cache/save@v4
213213
with:
214214
path: ${{ github.workspace }}/.numba_cache
215215
# Save cache with the current date (ENV set in numba_cache action)
@@ -249,7 +249,7 @@ jobs:
249249
run: python -m pytest -n logical --doctest-only --doctest-continue-on-failure
250250

251251
- name: Save new cache
252-
uses: actions/cache/save@v5
252+
uses: actions/cache/save@v4
253253
with:
254254
path: ${{ github.workspace }}/.numba_cache
255255
# Save cache with the current date (ENV set in numba_cache action)
@@ -289,7 +289,7 @@ jobs:
289289
run: python -m pytest aeon/testing/tests/ --enablethreading true -k "check_estimator_multithreading"
290290

291291
- name: Save new cache
292-
uses: actions/cache/save@v5
292+
uses: actions/cache/save@v4
293293
with:
294294
path: ${{ github.workspace }}/.numba_cache
295295
# Save cache with the current date (ENV set in numba_cache action)

.github/workflows/pr_precommit.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -35,7 +35,7 @@ jobs:
3535
python-version: "3.12"
3636

3737
- name: Get changed files
38-
uses: tj-actions/changed-files@v47.0.1
38+
uses: tj-actions/changed-files@v47.0.0
3939
id: changed-files
4040

4141
- name: List changed files

.github/workflows/release.yml

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -40,7 +40,7 @@ jobs:
4040
python -m build
4141
4242
- name: Store build files
43-
uses: actions/upload-artifact@v6
43+
uses: actions/upload-artifact@v5
4444
with:
4545
name: dist
4646
path: dist/*
@@ -78,7 +78,7 @@ jobs:
7878
with:
7979
swap-size-gb: 12
8080

81-
- uses: actions/download-artifact@v7
81+
- uses: actions/download-artifact@v6
8282
with:
8383
name: dist
8484
path: dist
@@ -130,7 +130,7 @@ jobs:
130130
id-token: write
131131

132132
steps:
133-
- uses: actions/download-artifact@v7
133+
- uses: actions/download-artifact@v6
134134
with:
135135
name: dist
136136
path: dist

.github/workflows/update_contributors.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -31,7 +31,7 @@ jobs:
3131
app-id: ${{ vars.PR_APP_ID }}
3232
private-key: ${{ secrets.PR_APP_KEY }}
3333

34-
- uses: peter-evans/create-pull-request@v8
34+
- uses: peter-evans/create-pull-request@v7
3535
with:
3636
token: ${{ steps.app-token.outputs.token }}
3737
commit-message: "Automated `CONTRIBUTORS.md` update"

.github/workflows/weekly_github_maintenance.yml

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,7 @@ jobs:
3030
private-key: ${{ secrets.PR_APP_KEY }}
3131

3232
- if: always()
33-
uses: peter-evans/create-pull-request@v8
33+
uses: peter-evans/create-pull-request@v7
3434
with:
3535
token: ${{ steps.app-token.outputs.token }}
3636
commit-message: "Automated `pre-commit` hook update"
@@ -67,7 +67,7 @@ jobs:
6767
env:
6868
GITHUB_TOKEN: ${{ steps.app-token.outputs.token }}
6969

70-
- uses: peter-evans/create-pull-request@v8
70+
- uses: peter-evans/create-pull-request@v7
7171
with:
7272
token: ${{ steps.app-token.outputs.token }}
7373
commit-message: "Automated teams tables update"
@@ -98,7 +98,7 @@ jobs:
9898
publish_results: true
9999

100100
- name: Upload artifact
101-
uses: actions/upload-artifact@v6
101+
uses: actions/upload-artifact@v5
102102
with:
103103
name: SARIF file
104104
path: results.sarif

.pre-commit-config.yaml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,7 @@ repos:
2727
args: [ "--create", "--python-folders", "aeon" ]
2828

2929
- repo: https://github.com/astral-sh/ruff-pre-commit
30-
rev: v0.14.10
30+
rev: v0.14.6
3131
hooks:
3232
- id: ruff
3333
args: [ "--fix"]
@@ -53,7 +53,7 @@ repos:
5353
args: [ "--max-line-length=88", "--extend-ignore=E203" ]
5454

5555
- repo: https://github.com/psf/black
56-
rev: 25.12.0
56+
rev: 25.11.0
5757
hooks:
5858
- id: black
5959
language_version: python3

aeon/classification/shapelet_based/tests/test_ls.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,13 @@
33
import numpy as np
44
import pytest
55

6+
try:
7+
import tensorflow as tf
8+
9+
tf.random.set_seed(1)
10+
except ImportError:
11+
pass
12+
613
from aeon.classification.shapelet_based import LearningShapeletClassifier
714
from aeon.testing.data_generation import make_example_3d_numpy
815
from aeon.utils.validation._dependencies import _check_soft_dependencies

aeon/networks/tests/test_deepar.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,13 @@
22

33
import pytest
44

5+
try:
6+
import tensorflow as tf
7+
8+
tf.random.set_seed(1)
9+
except ImportError:
10+
pass
11+
512
from aeon.networks import DeepARNetwork
613
from aeon.utils.validation._dependencies import _check_soft_dependencies
714

aeon/regression/deep_learning/tests/test_deep_regressor_base.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -6,6 +6,13 @@
66

77
import pytest
88

9+
try:
10+
import tensorflow as tf
11+
12+
tf.random.set_seed(1)
13+
except ImportError:
14+
pass
15+
916
from aeon.regression.deep_learning.base import BaseDeepRegressor
1017
from aeon.testing.data_generation import make_example_2d_numpy_collection
1118
from aeon.utils.validation._dependencies import _check_soft_dependencies

0 commit comments

Comments
 (0)