Skip to content

Commit 041e242

Browse files
committed
Parameter Fitter: Replace variadic argument in fitted_parameters with 'problem_type'
1 parent e34d66d commit 041e242

File tree

8 files changed

+11
-14
lines changed

8 files changed

+11
-14
lines changed

Orange/base.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -190,7 +190,8 @@ def active_preprocessors(self):
190190
self.preprocessors is not type(self).preprocessors):
191191
yield from type(self).preprocessors
192192

193-
def fitted_parameters(self, *args, **kwargs) -> list:
193+
# declared for derived classes, pylint: disable=unused-argument
194+
def fitted_parameters(self, problem_type: Union[str, Table, Domain]) -> list:
194195
return []
195196

196197
# pylint: disable=no-self-use

Orange/classification/random_forest.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -59,6 +59,6 @@ def __init__(self,
5959
super().__init__(preprocessors=preprocessors)
6060
self.params = vars()
6161

62-
def fitted_parameters(self) -> list[Learner.FittedParameter]:
62+
def fitted_parameters(self, _) -> list[Learner.FittedParameter]:
6363
return [self.FittedParameter("n_estimators", "Number of trees",
6464
"Trees", int, 1, None)]

Orange/modelling/randomforest.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -31,4 +31,4 @@ def fitted_parameters(
3131
self,
3232
problem_type: Union[str, Table, Domain]
3333
) -> list[Learner.FittedParameter]:
34-
return self.get_learner(problem_type).fitted_parameters()
34+
return self.get_learner(problem_type).fitted_parameters(problem_type)

Orange/regression/pls.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -255,7 +255,7 @@ def incompatibility_reason(self, domain):
255255
reason = "Only numeric target variables expected."
256256
return reason
257257

258-
def fitted_parameters(self) -> list[Learner.FittedParameter]:
258+
def fitted_parameters(self, _) -> list[Learner.FittedParameter]:
259259
return [self.FittedParameter("n_components", "Number of components",
260260
"Comp", int, 1, None)]
261261

Orange/regression/random_forest.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -58,6 +58,6 @@ def __init__(self,
5858
super().__init__(preprocessors=preprocessors)
5959
self.params = vars()
6060

61-
def fitted_parameters(self) -> list[Learner.FittedParameter]:
61+
def fitted_parameters(self, _) -> list[Learner.FittedParameter]:
6262
return [self.FittedParameter("n_estimators", "Number of trees",
6363
"Trees", int, 1, None)]

Orange/regression/tests/test_pls.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,7 @@ def table(rows, attr, variables):
2222

2323
class TestPLSRegressionLearner(unittest.TestCase):
2424
def test_fitted_parameters(self):
25-
fitted_parameters = PLSRegressionLearner().fitted_parameters()
25+
fitted_parameters = PLSRegressionLearner().fitted_parameters(None)
2626
self.assertIsInstance(fitted_parameters, list)
2727
self.assertEqual(len(fitted_parameters), 1)
2828

Orange/widgets/evaluate/owparameterfitter.py

Lines changed: 3 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -439,14 +439,10 @@ def _settings_changed(self):
439439

440440
@property
441441
def fitted_parameters(self) -> list:
442-
if not self._learner:
442+
if not self._learner \
443+
or isinstance(self._learner, Fitter) and not self._data:
443444
return []
444-
if isinstance(self._learner, Fitter):
445-
if not self._data:
446-
return []
447-
return self._learner.fitted_parameters(self._data)
448-
else:
449-
return self._learner.fitted_parameters()
445+
return self._learner.fitted_parameters(self._data)
450446

451447
@property
452448
def initial_parameters(self) -> dict:

Orange/widgets/evaluate/tests/test_owparameterfitter.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,7 @@
1919

2020

2121
class DummyLearner(PLSRegressionLearner):
22-
def fitted_parameters(self):
22+
def fitted_parameters(self, _):
2323
return [
2424
self.FittedParameter("n_components", "Foo", "foo", int, 5, None),
2525
self.FittedParameter("n_components", "Bar", "bar", int, 5, 10),

0 commit comments

Comments
 (0)