Skip to content

Commit cd2f729

Browse files
committed
Explicit int64/float64 dtype declaration
1 parent e1d1e45 commit cd2f729

2 files changed

Lines changed: 8 additions & 8 deletions

File tree

stumpy/maamp.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -499,10 +499,10 @@ def _maamp(
499499
QT_even = QT.copy()
500500
d = T.shape[0]
501501

502-
P = np.empty((d, range_stop - range_start), dtype=float)
503-
I = np.empty((d, range_stop - range_start), dtype=int)
504-
D = np.empty((d, k), dtype=float)
505-
D_prime = np.empty(k, dtype=float)
502+
P = np.empty((d, range_stop - range_start), dtype=np.float64)
503+
I = np.empty((d, range_stop - range_start), dtype=np.int64)
504+
D = np.empty((d, k), dtype=np.float64)
505+
D_prime = np.empty(k, dtype=np.float64)
506506
start_row_idx = 0
507507

508508
if include is not None:

stumpy/mstump.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -793,10 +793,10 @@ def _mstump(
793793
QT_even = QT.copy()
794794
d = T.shape[0]
795795

796-
P = np.empty((d, range_stop - range_start), dtype=float)
797-
I = np.empty((d, range_stop - range_start), dtype=int)
798-
D = np.empty((d, k), dtype=float)
799-
D_prime = np.empty(k, dtype=float)
796+
P = np.empty((d, range_stop - range_start), dtype=np.float64)
797+
I = np.empty((d, range_stop - range_start), dtype=np.int64)
798+
D = np.empty((d, k), dtype=np.float64)
799+
D_prime = np.empty(k, dtype=np.float64)
800800
start_row_idx = 0
801801

802802
if include is not None:

0 commit comments

Comments
 (0)