From 9d19442f5d57908a49936120b994038ca77473b1 Mon Sep 17 00:00:00 2001 From: dachengx Date: Thu, 8 Aug 2024 01:09:05 -0500 Subject: [PATCH] Debug, numba dose not understand how to extract dtype --- strax/processing/general.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/strax/processing/general.py b/strax/processing/general.py index b3b63b9e..5e371f04 100644 --- a/strax/processing/general.py +++ b/strax/processing/general.py @@ -82,9 +82,10 @@ def _overload_endtime(x): @numba.jit(nopython=True, nogil=True, cache=True) def diff(data): """Return time differences between items in data.""" + # we are sure that time is np.int64 if len(data) == 0: - return np.zeros(0, dtype=data.dtype["time"]) - results = np.zeros(len(data) - 1, dtype=data.dtype["time"]) + return np.zeros(0, dtype=np.int64) + results = np.zeros(len(data) - 1, dtype=np.int64) max_endtime = strax.endtime(data[0]) for i, (time, endtime) in enumerate(zip(data["time"][1:], strax.endtime(data)[:-1])): results[i] = time - max_endtime