|
1 | 1 | import numpy as np |
| 2 | +from skbase.utils.dependencies import _check_soft_dependencies |
2 | 3 |
|
3 | 4 | from hyperactive.base import BaseExperiment |
4 | 5 | from hyperactive.experiment.integrations._skl_metrics import _coerce_to_scorer_and_sign |
@@ -218,22 +219,43 @@ def _safe_index(self, obj, idx): |
218 | 219 | @classmethod |
219 | 220 | def get_test_params(cls, parameter_set="default"): |
220 | 221 | # 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 |
225 | 227 |
|
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 |
230 | 236 | X = None |
231 | 237 | y = None |
232 | 238 |
|
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 = { |
234 | 246 | "detector": DummyDetector() if DummyDetector is not None else None, |
235 | 247 | "X": X, |
236 | 248 | "y": y, |
| 249 | + "cv": 2, |
| 250 | + "scoring": None, |
| 251 | + "error_score": 0.0, |
| 252 | + "backend": "loky", |
| 253 | + "backend_params": {"n_jobs": 1}, |
237 | 254 | } |
238 | 255 |
|
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