Skip to content

Commit

Permalink
refactor(core): simplify, as steps can't be nested
Browse files Browse the repository at this point in the history
  • Loading branch information
deepyaman committed Sep 13, 2024
1 parent 6c8c815 commit 5e8c761
Showing 1 changed file with 1 addition and 14 deletions.
15 changes: 1 addition & 14 deletions ibis_ml/core.py
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,6 @@
import ibis
import ibis.expr.operations as ops
import ibis.expr.types as ir
from ibis.common.deferred import Deferred
from ibis.common.dispatch import lazy_singledispatch

if TYPE_CHECKING:
Expand Down Expand Up @@ -370,19 +369,7 @@ def get_params(self, deep=True) -> dict[str, Any]:
----------
.. [1] https://github.com/scikit-learn/scikit-learn/blob/626b460/sklearn/base.py#L145-L167
"""
out = {}
for key in self._get_param_names():
value = getattr(self, key)
if (
deep
and hasattr(value, "get_params")
# `hasattr()` always returns `True` for deferred objects
and not isinstance(value, (type, Deferred))
):
deep_items = value.get_params().items()
out.update((key + "__" + k, val) for k, val in deep_items)
out[key] = value
return out
return {key: getattr(self, key) for key in self._get_param_names()}

def __repr__(self) -> str:
return pprint.pformat(self)
Expand Down

0 comments on commit 5e8c761

Please sign in to comment.