Skip to content

Commit 7f083f3

Browse files
committed
Revert test_tsfresh.py: tsfresh works on CPU after numba-cuda removal
Now that numba-cuda is uninstalled on the CPU image, tsfresh/stumpy imports work without requiring libcudart.so. No need for the skip guard.
1 parent 786b9fd commit 7f083f3

1 file changed

Lines changed: 1 addition & 6 deletions

File tree

tests/test_tsfresh.py

Lines changed: 1 addition & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -3,13 +3,8 @@
33
import pandas as pd
44
import numpy as np
55

6-
try:
7-
from tsfresh import extract_features
8-
HAS_TSFRESH = True
9-
except ImportError:
10-
HAS_TSFRESH = False
6+
from tsfresh import extract_features
117

12-
@unittest.skipUnless(HAS_TSFRESH, 'tsfresh is not importable (may require CUDA libs)')
138
class TestTsFresh(unittest.TestCase):
149
def test_extract_feature(self):
1510
ts = pd.DataFrame({

0 commit comments

Comments
 (0)