Skip to content

Commit 7e4b945

Browse files
authored
Added gpu_aamp_stimp driver error
1 parent bd2a43c commit 7e4b945

2 files changed

Lines changed: 8 additions & 0 deletions

File tree

stumpy/__init__.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -46,6 +46,7 @@
4646
from .core import _gpu_mpdist_driver_not_found as gpu_mpdist # noqa: F401
4747
from .core import _gpu_aampdist_driver_not_found as gpu_aampdist # noqa: F401
4848
from .core import _gpu_stimp_driver_not_found as gpu_stimp # noqa: F401
49+
from .core import _gpu_aamp_stimp_driver_not_found as gpu_aamp_stimp # noqa: F401
4950
import ast
5051
import pathlib
5152

stumpy/core.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -199,6 +199,13 @@ def _gpu_stimp_driver_not_found(*args, **kwargs): # pragma: no cover
199199
driver_not_found()
200200

201201

202+
def _gpu_aamp_stimp_driver_not_found(*args, **kwargs): # pragma: no cover
203+
"""
204+
Dummy function to raise CudaSupportError driver not found error.
205+
"""
206+
driver_not_found()
207+
208+
202209
def get_pkg_name(): # pragma: no cover
203210
"""
204211
Return package name.

0 commit comments

Comments
 (0)