Skip to content

Commit 5c52f3d

Browse files
authored
Fixed #441 Add aamp stimp (#572)
* Made excl_zone optional in _multi_distance_profile funcs * Added initial aamp stimp implementations * Added correct normalization factor for ED * Ran black, added unit test shims * Added p-norm * Completed unit tests and checks
1 parent d59c286 commit 5c52f3d

16 files changed

Lines changed: 1433 additions & 371 deletions

stumpy/__init__.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -24,6 +24,7 @@
2424
from .snippets import snippets # noqa: F401
2525
from .aampdist_snippets import aampdist_snippets # noqa: F401
2626
from .stimp import stimp, stimped # noqa: F401
27+
from .aamp_stimp import aamp_stimp, aamp_stimped # noqa: F401
2728
from numba import cuda
2829

2930
if cuda.is_available():
@@ -34,6 +35,7 @@
3435
from .gpu_mpdist import gpu_mpdist # noqa: F401
3536
from .gpu_aampdist import gpu_aampdist # noqa: F401
3637
from .gpu_stimp import gpu_stimp # noqa: F401
38+
from .gpu_aamp_stimp import gpu_aamp_stimp # noqa: F401
3739
else: # pragma: no cover
3840
from .core import _gpu_stump_driver_not_found as gpu_stump # noqa: F401
3941
from .core import _gpu_aamp_driver_not_found as gpu_aamp # noqa: F401

0 commit comments

Comments
 (0)