Commit 786b9fd
committed
Uninstall numba-cuda on CPU image to fix tsfresh import
Instead of just skipping tsfresh tests on CPU, uninstall numba-cuda
from the Colab base image on CPU builds. This lets tsfresh/stumpy
work correctly on CPU without CUDA dependencies.
On GPU, numba-cuda is upgraded to v0.30.0 for NumPy 2.4 support.1 parent 6d4b24e commit 786b9fd
1 file changed
Lines changed: 5 additions & 0 deletions
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
43 | 43 | | |
44 | 44 | | |
45 | 45 | | |
| 46 | + | |
| 47 | + | |
| 48 | + | |
| 49 | + | |
| 50 | + | |
46 | 51 | | |
47 | 52 | | |
48 | 53 | | |
| |||
0 commit comments