Skip to content

Commit 93d5079

Browse files
[MNT] Remove REDCOMETs from testing exclusion list (aeon-toolkit#2630)
* remove excluded estimators * redcomets fix
1 parent eb761f5 commit 93d5079

File tree

2 files changed

+2
-5
lines changed

2 files changed

+2
-5
lines changed

aeon/classification/dictionary_based/_redcomets.py

+2-4
Original file line numberDiff line numberDiff line change
@@ -255,8 +255,7 @@ def _build_univariate_ensemble(self, X, y):
255255
sfa_clfs = []
256256
for sfa in sfa_transforms:
257257
sfa.fit(X_smote, y_smote)
258-
sfa_dics = sfa.transform_words(X_smote)
259-
X_sfa = sfa_dics[:, 0, :]
258+
X_sfa = sfa.transform_words(X_smote)[0]
260259

261260
rf = RandomForestClassifier(
262261
n_estimators=self.n_trees,
@@ -417,8 +416,7 @@ def _predict_proba_unvivariate(self, X) -> np.ndarray:
417416
pred_mat = np.zeros((X.shape[0], self.n_classes_))
418417

419418
for sfa, (rf, weight) in zip(self.sfa_transforms, self.sfa_clfs):
420-
sfa_dics = sfa.transform_words(X)
421-
X_sfa = sfa_dics[:, 0, :]
419+
X_sfa = sfa.transform_words(X)[0]
422420

423421
rf_pred_mat = rf.predict_proba(X_sfa)
424422

aeon/testing/testing_config.py

-1
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,6 @@
2424

2525
# exclude estimators here for short term fixes
2626
EXCLUDE_ESTIMATORS = [
27-
"REDCOMETS",
2827
"HydraTransformer", # returns a pytorch Tensor
2928
]
3029

0 commit comments

Comments
 (0)