Skip to content

Commit b397427

Browse files
committed
soft check
1 parent 40c8022 commit b397427

1 file changed

Lines changed: 32 additions & 10 deletions

File tree

src/hyperactive/experiment/integrations/sktime_detector.py

Lines changed: 32 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
import numpy as np
2+
from skbase.utils.dependencies import _check_soft_dependencies
23

34
from hyperactive.base import BaseExperiment
45
from hyperactive.experiment.integrations._skl_metrics import _coerce_to_scorer_and_sign
@@ -218,22 +219,43 @@ def _safe_index(self, obj, idx):
218219
@classmethod
219220
def get_test_params(cls, parameter_set="default"):
220221
# Return testing parameter settings for the skbase object.
221-
try:
222-
from sktime.annotation.dummy import DummyDetector
223-
except Exception:
224-
DummyDetector = None
222+
if _check_soft_dependencies("sktime", severity="none"):
223+
try:
224+
from sktime.annotation.dummy import DummyDetector
225+
except Exception:
226+
DummyDetector = None
225227

226-
try:
227-
from sktime.datasets import load_unit_test
228-
X, y = load_unit_test(return_X_y=True, return_type="pd-multiindex")
229-
except Exception:
228+
try:
229+
from sktime.datasets import load_unit_test
230+
X, y = load_unit_test(return_X_y=True, return_type="pd-multiindex")
231+
except Exception:
232+
X = None
233+
y = None
234+
else:
235+
DummyDetector = None
230236
X = None
231237
y = None
232238

233-
params0 = {
239+
params_default = {
240+
"detector": DummyDetector() if DummyDetector is not None else None,
241+
"X": X,
242+
"y": y,
243+
}
244+
245+
params_more = {
234246
"detector": DummyDetector() if DummyDetector is not None else None,
235247
"X": X,
236248
"y": y,
249+
"cv": 2,
250+
"scoring": None,
251+
"error_score": 0.0,
252+
"backend": "loky",
253+
"backend_params": {"n_jobs": 1},
237254
}
238255

239-
return [params0]
256+
if parameter_set == "default":
257+
return [params_default]
258+
elif parameter_set == "more_params":
259+
return [params_more]
260+
else:
261+
return [params_default]

0 commit comments

Comments
 (0)