Skip to content

Commit c2036fc

Browse files
MatthewMiddlehurstpattplatt
authored andcommitted
correct import and tag
1 parent 1d3b454 commit c2036fc

File tree

2 files changed

+9
-3
lines changed

2 files changed

+9
-3
lines changed

aeon/anomaly_detection/whole_series/_outlier_detection.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
"""Basic outlier detection classifier."""
22

3-
from aeon.anomaly_detection import IsolationForest
3+
from sklearn.ensemble import IsolationForest
4+
45
from aeon.anomaly_detection.whole_series.base import BaseCollectionAnomalyDetector
56
from aeon.base._base import _clone_estimator
67

@@ -32,4 +33,4 @@ def _predict(self, X):
3233

3334
@classmethod
3435
def _get_test_params(cls, parameter_set="default"):
35-
return {"estimator": IsolationForest()}
36+
return {"estimator": IsolationForest(n_estimators=3)}

aeon/utils/tags/_tags.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -138,7 +138,12 @@ class : identifier for the base class of objects this tag applies to
138138
"point belongs to.",
139139
},
140140
"requires_y": {
141-
"class": ["transformer", "anomaly-detector", "segmenter", "similarity-search"],
141+
"class": [
142+
"transformer",
143+
"anomaly-detector",
144+
"collection_anomaly_detector",
145+
"segmenter",
146+
],
142147
"type": "bool",
143148
"description": "Does this estimator require y to be passed in its methods?",
144149
},

0 commit comments

Comments
 (0)