From 45a30bcd6a7d631b7b8e20a32b7d6ee7411c51c2 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Franz=20Kir=C3=A1ly?= Date: Wed, 3 Jan 2024 18:13:03 +0100 Subject: [PATCH] [ENH] add tags to `SklearnProbaReg` (#168) This PR adds tags in `SklearnProbaReg`, otherwise one of the tests complains. Should not matter due to inheritance, and odd that it did not complain when merged. --- skpro/regression/adapters/sklearn/_sklearn_proba.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/skpro/regression/adapters/sklearn/_sklearn_proba.py b/skpro/regression/adapters/sklearn/_sklearn_proba.py index 0f79e973..aeedb17a 100644 --- a/skpro/regression/adapters/sklearn/_sklearn_proba.py +++ b/skpro/regression/adapters/sklearn/_sklearn_proba.py @@ -25,7 +25,10 @@ class SklearnProbaReg(BaseProbaRegressor): Estimator to wrap, must have ``predict`` with ``return_std`` argument. """ - _tags = {} + _tags = { + "capability:multioutput": False, + "capability:missing": True, + } def __init__(self, estimator): self.estimator = estimator