Skip to content

Commit d80f76d

Browse files
authored
remove numba typing (#2858)
1 parent f110796 commit d80f76d

File tree

2 files changed

+1
-16
lines changed

2 files changed

+1
-16
lines changed

aeon/transformations/collection/convolution_based/_minirocket.py

Lines changed: 1 addition & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -230,16 +230,14 @@ def _static_fit(X, n_features=10_000, max_dilations_per_kernel=32, seed=None):
230230
)
231231

232232

233-
@vectorize("float32(float32,float32)", nopython=True, cache=True)
233+
@vectorize(nopython=True, cache=True)
234234
def _PPV(a, b):
235235
if a > b:
236236
return 1
237237
return 0
238238

239239

240240
@njit(
241-
"float32[:,:](float32[:,:],Tuple((int32[:],int32[:],int32[:],int32[:],float32["
242-
":])), int32[:,:])",
243241
fastmath=True,
244242
parallel=True,
245243
cache=True,
@@ -305,8 +303,6 @@ def _static_transform_uni(X, parameters, indices):
305303

306304

307305
@njit(
308-
"float32[:,:](float32[:,:,:],Tuple((int32[:],int32[:],int32[:],int32[:],float32["
309-
":])), int32[:,:])",
310306
fastmath=True,
311307
parallel=True,
312308
cache=True,
@@ -384,8 +380,6 @@ def _static_transform_multi(X, parameters, indices):
384380

385381

386382
@njit(
387-
"float32[:](float32[:,:,:],int32[:],int32[:],int32[:],int32[:],float32[:],"
388-
"int32[:,:],optional(int32))", # noqa
389383
fastmath=True,
390384
parallel=False,
391385
cache=True,

aeon/transformations/collection/convolution_based/_multirocket.py

Lines changed: 0 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -279,8 +279,6 @@ def _fit_multivariate(self, X):
279279

280280

281281
@njit(
282-
"float32[:,:](float32[:,:],float32[:,:],Tuple((int32[:],int32[:],float32[:])),"
283-
"Tuple((int32[:],int32[:],float32[:])),int32, int32[:,:],optional(int32))",
284282
fastmath=True,
285283
parallel=True,
286284
cache=True,
@@ -553,10 +551,6 @@ def _transform_uni(
553551

554552

555553
@njit(
556-
"float32[:,:](float32[:,:,:],float32[:,:,:],"
557-
"Tuple((int32[:],int32[:],int32[:],int32[:],float32[:])),"
558-
"Tuple((int32[:],int32[:],int32[:],int32[:],float32[:])),int32, int32[:,:],"
559-
"optional(int32))",
560554
fastmath=True,
561555
parallel=True,
562556
cache=True,
@@ -875,7 +869,6 @@ def _transform_multi(
875869

876870

877871
@njit(
878-
"float32[:](float32[:,:],int32[:],int32[:],float32[:], int32[:,:],optional(int32))",
879872
fastmath=True,
880873
parallel=False,
881874
cache=True,
@@ -945,8 +938,6 @@ def _fit_biases_univariate(
945938

946939

947940
@njit(
948-
"float32[:](float32[:,:,:],int32[:],int32[:],int32[:],int32[:],float32[:], "
949-
"int32[:,:],optional(int32))",
950941
fastmath=True,
951942
parallel=False,
952943
cache=True,

0 commit comments

Comments
 (0)