Skip to content

Commit

Permalink
[ENH] add tags to SklearnProbaReg (#168)
Browse files Browse the repository at this point in the history
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.
  • Loading branch information
fkiraly committed Jan 3, 2024
1 parent b483e46 commit 45a30bc
Showing 1 changed file with 4 additions and 1 deletion.
5 changes: 4 additions & 1 deletion skpro/regression/adapters/sklearn/_sklearn_proba.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit 45a30bc

Please sign in to comment.