scikit-base 0.4.6__py3-none-any.whl → 0.5.1__py3-none-any.whl

skbase/tests/test_baseestimator.py

@@ -1,130 +1,130 @@
-# -*- coding: utf-8 -*-
-# copyright: skbase developers, BSD-3-Clause License (see LICENSE file)
-"""Tests for BaseEstimator class.
-
-tests in this module:
-
-    test_baseestimator_inheritance - Test BaseEstimator inherits from BaseObject.
-    test_has_is_fitted - Test that BaseEstimator has is_fitted interface.
-    test_has_check_is_fitted - Test that BaseEstimator has check_is_fitted inteface.
-    test_is_fitted  - Test that is_fitted property returns _is_fitted as expected.
-    test_check_is_fitted_raises_error_when_unfitted - Test check_is_fitted raises error.
-"""
-
-__author__ = ["fkiraly", "RNKuhns"]
-import inspect
-from copy import deepcopy
-
-import pytest
-
-from skbase._exceptions import NotFittedError
-from skbase.base import BaseEstimator, BaseObject
-
-
-@pytest.fixture
-def fixture_estimator():
-    """Pytest fixture of BaseEstimator class."""
-    return BaseEstimator
-
-
-@pytest.fixture
-def fixture_estimator_instance():
-    """Pytest fixture of BaseEstimator instance."""
-    return BaseEstimator()
-
-
-def test_baseestimator_inheritance(fixture_estimator, fixture_estimator_instance):
-    """Check BaseEstimator correctly inherits from BaseObject."""
-    estimator_is_subclass_of_baseobejct = issubclass(fixture_estimator, BaseObject)
-    estimator_instance_is_baseobject_instance = isinstance(
-        fixture_estimator_instance, BaseObject
-    )
-    assert (
-        estimator_is_subclass_of_baseobejct
-        and estimator_instance_is_baseobject_instance
-    ), "`BaseEstimator` does not correctly inherit from `BaseObject`."
-
-
-def test_has_is_fitted(fixture_estimator_instance):
-    """Test BaseEstimator has `is_fitted` property."""
-    has_private_is_fitted = hasattr(fixture_estimator_instance, "_is_fitted")
-    has_is_fitted = hasattr(fixture_estimator_instance, "is_fitted")
-    assert (
-        has_private_is_fitted and has_is_fitted
-    ), "BaseEstimator does not have `is_fitted` property;"
-
-
-def test_has_check_is_fitted(fixture_estimator_instance):
-    """Test BaseEstimator has `check_is_fitted` method."""
-    has_check_is_fitted = hasattr(fixture_estimator_instance, "check_is_fitted")
-    is_method = inspect.ismethod(fixture_estimator_instance.check_is_fitted)
-    assert (
-        has_check_is_fitted and is_method
-    ), "`BaseEstimator` does not have `check_is_fitted` method."
-
-
-def test_is_fitted(fixture_estimator_instance):
-    """Test BaseEstimator `is_fitted` property returns expected value."""
-    expected_value_unfitted = (
-        fixture_estimator_instance.is_fitted == fixture_estimator_instance._is_fitted
-    )
-    assert (
-        expected_value_unfitted
-    ), "`BaseEstimator` property `is_fitted` does not return `_is_fitted` value."
-
-
-def test_check_is_fitted_raises_error_when_unfitted(fixture_estimator_instance):
-    """Test BaseEstimator `check_is_fitted` method raises an error."""
-    name = fixture_estimator_instance.__class__.__name__
-    match = f"This instance of {name} has not been fitted yet. Please call `fit` first."
-    with pytest.raises(NotFittedError, match=match):
-        fixture_estimator_instance.check_is_fitted()
-
-    fixture_estimator_instance._is_fitted = True
-    assert fixture_estimator_instance.check_is_fitted() is None
-
-
-class FittableCompositionDummy(BaseEstimator):
-    """Potentially composite object, for testing."""
-
-    def __init__(self, foo, bar=84):
-        self.foo = foo
-        self.foo_ = deepcopy(foo)
-        self.bar = bar
-
-    def fit(self):
-        """Fit, dummy."""
-        if hasattr(self.foo_, "fit"):
-            self.foo_.fit()
-        self._is_fitted = True
-
-
-def test_get_fitted_params():
-    """Tests fitted parameter retrieval.
-
-    Raises
-    ------
-    AssertionError if logic behind get_fitted_params is incorrect, logic tested:
-        calling get_fitted_params on a non-composite fittable returns the fitted param
-        calling get_fitted_params on a composite returns all nested params
-    """
-    non_composite = FittableCompositionDummy(foo=42)
-    composite = FittableCompositionDummy(foo=deepcopy(non_composite))
-
-    non_composite.fit()
-    composite.fit()
-
-    non_comp_f_params = non_composite.get_fitted_params()
-    comp_f_params = composite.get_fitted_params()
-    comp_f_params_shallow = composite.get_fitted_params(deep=False)
-
-    assert isinstance(non_comp_f_params, dict)
-    assert set(non_comp_f_params.keys()) == {"foo"}
-
-    assert isinstance(comp_f_params, dict)
-    assert set(comp_f_params) == {"foo", "foo__foo"}
-    assert set(comp_f_params_shallow) == {"foo"}
-    assert comp_f_params["foo"] is composite.foo_
-    assert comp_f_params["foo"] is not composite.foo
-    assert comp_f_params_shallow["foo"] is composite.foo_
-    assert comp_f_params_shallow["foo"] is not composite.foo
+# -*- coding: utf-8 -*-
+# copyright: skbase developers, BSD-3-Clause License (see LICENSE file)
+"""Tests for BaseEstimator class.
+
+tests in this module:
+
+    test_baseestimator_inheritance - Test BaseEstimator inherits from BaseObject.
+    test_has_is_fitted - Test that BaseEstimator has is_fitted interface.
+    test_has_check_is_fitted - Test that BaseEstimator has check_is_fitted inteface.
+    test_is_fitted  - Test that is_fitted property returns _is_fitted as expected.
+    test_check_is_fitted_raises_error_when_unfitted - Test check_is_fitted raises error.
+"""
+
+__author__ = ["fkiraly", "RNKuhns"]
+import inspect
+from copy import deepcopy
+
+import pytest
+
+from skbase._exceptions import NotFittedError
+from skbase.base import BaseEstimator, BaseObject
+
+
+@pytest.fixture
+def fixture_estimator():
+    """Pytest fixture of BaseEstimator class."""
+    return BaseEstimator
+
+
+@pytest.fixture
+def fixture_estimator_instance():
+    """Pytest fixture of BaseEstimator instance."""
+    return BaseEstimator()
+
+
+def test_baseestimator_inheritance(fixture_estimator, fixture_estimator_instance):
+    """Check BaseEstimator correctly inherits from BaseObject."""
+    estimator_is_subclass_of_baseobejct = issubclass(fixture_estimator, BaseObject)
+    estimator_instance_is_baseobject_instance = isinstance(
+        fixture_estimator_instance, BaseObject
+    )
+    assert (
+        estimator_is_subclass_of_baseobejct
+        and estimator_instance_is_baseobject_instance
+    ), "`BaseEstimator` does not correctly inherit from `BaseObject`."
+
+
+def test_has_is_fitted(fixture_estimator_instance):
+    """Test BaseEstimator has `is_fitted` property."""
+    has_private_is_fitted = hasattr(fixture_estimator_instance, "_is_fitted")
+    has_is_fitted = hasattr(fixture_estimator_instance, "is_fitted")
+    assert (
+        has_private_is_fitted and has_is_fitted
+    ), "BaseEstimator does not have `is_fitted` property;"
+
+
+def test_has_check_is_fitted(fixture_estimator_instance):
+    """Test BaseEstimator has `check_is_fitted` method."""
+    has_check_is_fitted = hasattr(fixture_estimator_instance, "check_is_fitted")
+    is_method = inspect.ismethod(fixture_estimator_instance.check_is_fitted)
+    assert (
+        has_check_is_fitted and is_method
+    ), "`BaseEstimator` does not have `check_is_fitted` method."
+
+
+def test_is_fitted(fixture_estimator_instance):
+    """Test BaseEstimator `is_fitted` property returns expected value."""
+    expected_value_unfitted = (
+        fixture_estimator_instance.is_fitted == fixture_estimator_instance._is_fitted
+    )
+    assert (
+        expected_value_unfitted
+    ), "`BaseEstimator` property `is_fitted` does not return `_is_fitted` value."
+
+
+def test_check_is_fitted_raises_error_when_unfitted(fixture_estimator_instance):
+    """Test BaseEstimator `check_is_fitted` method raises an error."""
+    name = fixture_estimator_instance.__class__.__name__
+    match = f"This instance of {name} has not been fitted yet. Please call `fit` first."
+    with pytest.raises(NotFittedError, match=match):
+        fixture_estimator_instance.check_is_fitted()
+
+    fixture_estimator_instance._is_fitted = True
+    assert fixture_estimator_instance.check_is_fitted() is None
+
+
+class FittableCompositionDummy(BaseEstimator):
+    """Potentially composite object, for testing."""
+
+    def __init__(self, foo, bar=84):
+        self.foo = foo
+        self.foo_ = deepcopy(foo)
+        self.bar = bar
+
+    def fit(self):
+        """Fit, dummy."""
+        if hasattr(self.foo_, "fit"):
+            self.foo_.fit()
+        self._is_fitted = True
+
+
+def test_get_fitted_params():
+    """Tests fitted parameter retrieval.
+
+    Raises
+    ------
+    AssertionError if logic behind get_fitted_params is incorrect, logic tested:
+        calling get_fitted_params on a non-composite fittable returns the fitted param
+        calling get_fitted_params on a composite returns all nested params
+    """
+    non_composite = FittableCompositionDummy(foo=42)
+    composite = FittableCompositionDummy(foo=deepcopy(non_composite))
+
+    non_composite.fit()
+    composite.fit()
+
+    non_comp_f_params = non_composite.get_fitted_params()
+    comp_f_params = composite.get_fitted_params()
+    comp_f_params_shallow = composite.get_fitted_params(deep=False)
+
+    assert isinstance(non_comp_f_params, dict)
+    assert set(non_comp_f_params.keys()) == {"foo"}
+
+    assert isinstance(comp_f_params, dict)
+    assert set(comp_f_params) == {"foo", "foo__foo"}
+    assert set(comp_f_params_shallow) == {"foo"}
+    assert comp_f_params["foo"] is composite.foo_
+    assert comp_f_params["foo"] is not composite.foo
+    assert comp_f_params_shallow["foo"] is composite.foo_
+    assert comp_f_params_shallow["foo"] is not composite.foo

skbase/tests/test_exceptions.py

@@ -1,23 +1,23 @@
-# -*- coding: utf-8 -*-
-"""Tests for skbase exceptions.
-
-tests in this module:
-
-    test_exceptions_raise_error - Test that skbase exceptions raise expected error.
-"""
-from typing import List
-
-import pytest
-
-from skbase._exceptions import FixtureGenerationError, NotFittedError
-
-__author__: List[str] = ["RNKuhns"]
-
-ALL_EXCEPTIONS = [FixtureGenerationError, NotFittedError]
-
-
-@pytest.mark.parametrize("skbase_exception", ALL_EXCEPTIONS)
-def test_exceptions_raise_error(skbase_exception):
-    """Test that skbase exceptions raise an error as expected."""
-    with pytest.raises(skbase_exception):
-        raise skbase_exception()
+# -*- coding: utf-8 -*-
+"""Tests for skbase exceptions.
+
+tests in this module:
+
+    test_exceptions_raise_error - Test that skbase exceptions raise expected error.
+"""
+from typing import List
+
+import pytest
+
+from skbase._exceptions import FixtureGenerationError, NotFittedError
+
+__author__: List[str] = ["RNKuhns"]
+
+ALL_EXCEPTIONS = [FixtureGenerationError, NotFittedError]
+
+
+@pytest.mark.parametrize("skbase_exception", ALL_EXCEPTIONS)
+def test_exceptions_raise_error(skbase_exception):
+    """Test that skbase exceptions raise an error as expected."""
+    with pytest.raises(skbase_exception):
+        raise skbase_exception()

skbase/tests/test_meta.py

@@ -1,131 +1,170 @@
-# -*- coding: utf-8 -*-
-# copyright: skbase developers, BSD-3-Clause License (see LICENSE file)
-"""Tests for BaseMetaObject and BaseMetaEstimator mixins.
-
-tests in this module:
-
-
-"""
-
-__author__ = ["RNKuhns"]
-import inspect
-
-import pytest
-
-from skbase._exceptions import NotFittedError
-from skbase.base import BaseEstimator, BaseObject
-from skbase.base._meta import (
-    BaseMetaEstimator,
-    BaseMetaObject,
-    _MetaObjectMixin,
-    _MetaTagLogicMixin,
-)
-
-
-class MetaObjectTester(BaseMetaObject):
-    """Class to test meta object functionality."""
-
-    def __init__(self, a=7, b="something", c=None, steps=None):
-        self.a = a
-        self.b = b
-        self.c = c
-        self.steps = steps
-
-
-class MetaEstimatorTester(BaseMetaEstimator):
-    """Class to test meta estimator functionality."""
-
-    def __init__(self, a=7, b="something", c=None, steps=None):
-        self.a = a
-        self.b = b
-        self.c = c
-        self.steps = steps
-
-
-@pytest.fixture
-def fixture_metaestimator_instance():
-    return BaseMetaEstimator()
-
-
-@pytest.fixture
-def fixture_meta_object():
-    return MetaObjectTester()
-
-
-@pytest.fixture
-def fixture_meta_estimator():
-    return MetaEstimatorTester()
-
-
-def test_is_composit_returns_true(fixture_meta_object, fixture_meta_estimator):
-    """Test that `is_composite` method returns True."""
-    msg = "`is_composite` should always be True for subclasses of "
-    assert fixture_meta_object.is_composite() is True, msg + "`BaseMetaObject`."
-    assert fixture_meta_estimator.is_composite() is True, msg + "`BaseMetaEstimator`."
-
-
-def test_basemetaestimator_inheritance(fixture_metaestimator_instance):
-    """Check BaseMetaEstimator correctly inherits from BaseEstimator and BaseObject."""
-    estimator_is_subclass_of_baseobejct = issubclass(BaseMetaEstimator, BaseObject)
-    estimator_instance_is_baseobject_instance = isinstance(
-        fixture_metaestimator_instance, BaseObject
-    )
-
-    # Verify that BaseMetaEstimator is an estimator
-    assert (
-        estimator_is_subclass_of_baseobejct
-        and estimator_instance_is_baseobject_instance
-    ), "`BaseMetaEstimator` not correctly subclassing `BaseEstimator` and `BaseObject`."
-
-    # Verify expected MRO inherittence order
-    assert BaseMetaEstimator.__mro__[:-2] == (
-        BaseMetaEstimator,
-        _MetaObjectMixin,
-        _MetaTagLogicMixin,
-        BaseEstimator,
-        BaseObject,
-    ), "`BaseMetaEstimator` has incorrect mro."
-
-
-def test_basemetaestimator_has_is_fitted(fixture_metaestimator_instance):
-    """Test BaseEstimator has `is_fitted` property."""
-    has_private_is_fitted = hasattr(fixture_metaestimator_instance, "_is_fitted")
-    has_is_fitted = hasattr(fixture_metaestimator_instance, "is_fitted")
-    assert (
-        has_private_is_fitted and has_is_fitted
-    ), "`BaseMetaEstimator` does not have `is_fitted` property or `_is_fitted` attr."
-
-
-def test_basemetaestimator_has_check_is_fitted(fixture_metaestimator_instance):
-    """Test BaseEstimator has `check_is_fitted` method."""
-    has_check_is_fitted = hasattr(fixture_metaestimator_instance, "check_is_fitted")
-    is_method = inspect.ismethod(fixture_metaestimator_instance.check_is_fitted)
-    assert (
-        has_check_is_fitted and is_method
-    ), "`BaseMetaEstimator` does not have `check_is_fitted` method."
-
-
-@pytest.mark.parametrize("is_fitted_value", (True, False))
-def test_basemetaestimator_is_fitted(fixture_metaestimator_instance, is_fitted_value):
-    """Test BaseEstimator `is_fitted` property returns expected value."""
-    fixture_metaestimator_instance._is_fitted = is_fitted_value
-    expected_value_unfitted = (
-        fixture_metaestimator_instance.is_fitted
-        == fixture_metaestimator_instance._is_fitted
-    )
-    assert (
-        expected_value_unfitted
-    ), "`BaseMetaEstimator` property `is_fitted` does not return `_is_fitted` value."
-
-
-def test_basemetaestimator_check_is_fitted_raises_error_when_unfitted(
-    fixture_metaestimator_instance,
-):
-    """Test BaseEstimator `check_is_fitted` method raises an error."""
-    name = fixture_metaestimator_instance.__class__.__name__
-    match = f"This instance of {name} has not been fitted yet. Please call `fit` first."
-    with pytest.raises(NotFittedError, match=match):
-        fixture_metaestimator_instance.check_is_fitted()
-
-    fixture_metaestimator_instance._is_fitted = True
-    assert fixture_metaestimator_instance.check_is_fitted() is None
+# -*- coding: utf-8 -*-
+# copyright: skbase developers, BSD-3-Clause License (see LICENSE file)
+"""Tests for BaseMetaObject and BaseMetaEstimator mixins."""
+
+__author__ = ["RNKuhns", "fkiraly"]
+import inspect
+
+import pytest
+
+from skbase._exceptions import NotFittedError
+from skbase.base import BaseEstimator, BaseObject
+from skbase.base._meta import (
+    BaseMetaEstimator,
+    BaseMetaObject,
+    _MetaObjectMixin,
+    _MetaTagLogicMixin,
+)
+
+
+class MetaObjectTester(BaseMetaObject):
+    """Class to test meta-object functionality."""
+
+    def __init__(self, a=7, b="something", c=None, steps=None):
+        self.a = a
+        self.b = b
+        self.c = c
+        self.steps = steps
+        super().__init__()
+
+
+class MetaEstimatorTester(BaseMetaEstimator):
+    """Class to test meta-estimator functionality."""
+
+    def __init__(self, a=7, b="something", c=None, steps=None):
+        self.a = a
+        self.b = b
+        self.c = c
+        self.steps = steps
+        super().__init__()
+
+
+class ComponentDummy(BaseObject):
+    """Class to use as components in meta-estimator."""
+
+    def __init__(self, a=7, b="something"):
+        self.a = a
+        self.b = b
+        super().__init__()
+
+
+@pytest.fixture
+def fixture_metaestimator_instance():
+    """BaseMetaEstimator instance fixture."""
+    return BaseMetaEstimator()
+
+
+@pytest.fixture
+def fixture_meta_object():
+    """MetaObjectTester instance fixture."""
+    return MetaObjectTester()
+
+
+@pytest.fixture
+def fixture_meta_estimator():
+    """MetaEstimatorTester instance fixture."""
+    return MetaEstimatorTester()
+
+
+def test_is_composit_returns_true(fixture_meta_object, fixture_meta_estimator):
+    """Test that `is_composite` method returns True."""
+    msg = "`is_composite` should always be True for subclasses of "
+    assert fixture_meta_object.is_composite() is True, msg + "`BaseMetaObject`."
+    assert fixture_meta_estimator.is_composite() is True, msg + "`BaseMetaEstimator`."
+
+
+def test_basemetaestimator_inheritance(fixture_metaestimator_instance):
+    """Check BaseMetaEstimator correctly inherits from BaseEstimator and BaseObject."""
+    estimator_is_subclass_of_baseobejct = issubclass(BaseMetaEstimator, BaseObject)
+    estimator_instance_is_baseobject_instance = isinstance(
+        fixture_metaestimator_instance, BaseObject
+    )
+
+    # Verify that BaseMetaEstimator is an estimator
+    assert (
+        estimator_is_subclass_of_baseobejct
+        and estimator_instance_is_baseobject_instance
+    ), "`BaseMetaEstimator` not correctly subclassing `BaseEstimator` and `BaseObject`."
+
+    # Verify expected MRO inherittence order
+    assert BaseMetaEstimator.__mro__[:-2] == (
+        BaseMetaEstimator,
+        _MetaObjectMixin,
+        _MetaTagLogicMixin,
+        BaseEstimator,
+        BaseObject,
+    ), "`BaseMetaEstimator` has incorrect mro."
+
+
+def test_basemetaestimator_has_is_fitted(fixture_metaestimator_instance):
+    """Test BaseEstimator has `is_fitted` property."""
+    has_private_is_fitted = hasattr(fixture_metaestimator_instance, "_is_fitted")
+    has_is_fitted = hasattr(fixture_metaestimator_instance, "is_fitted")
+    assert (
+        has_private_is_fitted and has_is_fitted
+    ), "`BaseMetaEstimator` does not have `is_fitted` property or `_is_fitted` attr."
+
+
+def test_basemetaestimator_has_check_is_fitted(fixture_metaestimator_instance):
+    """Test BaseEstimator has `check_is_fitted` method."""
+    has_check_is_fitted = hasattr(fixture_metaestimator_instance, "check_is_fitted")
+    is_method = inspect.ismethod(fixture_metaestimator_instance.check_is_fitted)
+    assert (
+        has_check_is_fitted and is_method
+    ), "`BaseMetaEstimator` does not have `check_is_fitted` method."
+
+
+@pytest.mark.parametrize("is_fitted_value", (True, False))
+def test_basemetaestimator_is_fitted(fixture_metaestimator_instance, is_fitted_value):
+    """Test BaseEstimator `is_fitted` property returns expected value."""
+    fixture_metaestimator_instance._is_fitted = is_fitted_value
+    expected_value_unfitted = (
+        fixture_metaestimator_instance.is_fitted
+        == fixture_metaestimator_instance._is_fitted
+    )
+    assert (
+        expected_value_unfitted
+    ), "`BaseMetaEstimator` property `is_fitted` does not return `_is_fitted` value."
+
+
+def test_basemetaestimator_check_is_fitted_raises_error_when_unfitted(
+    fixture_metaestimator_instance,
+):
+    """Test BaseEstimator `check_is_fitted` method raises an error."""
+    name = fixture_metaestimator_instance.__class__.__name__
+    match = f"This instance of {name} has not been fitted yet. Please call `fit` first."
+    with pytest.raises(NotFittedError, match=match):
+        fixture_metaestimator_instance.check_is_fitted()
+
+    fixture_metaestimator_instance._is_fitted = True
+    assert fixture_metaestimator_instance.check_is_fitted() is None
+
+
+@pytest.mark.parametrize("long_steps", (True, False))
+def test_metaestimator_composite(long_steps):
+    """Test composite meta-estimator functionality."""
+    if long_steps:
+        steps = [("foo", ComponentDummy(42)), ("bar", ComponentDummy(24))]
+    else:
+        steps = [("foo", ComponentDummy(42), 123), ("bar", ComponentDummy(24), 321)]
+
+    meta_est = MetaEstimatorTester(steps=steps)
+
+    meta_est_params = meta_est.get_params()
+    assert isinstance(meta_est_params, dict)
+    expected_keys = [
+        "a",
+        "b",
+        "c",
+        "steps",
+        "foo",
+        "bar",
+        "foo__a",
+        "foo__b",
+        "bar__a",
+        "bar__b",
+    ]
+    assert set(meta_est_params.keys()) == set(expected_keys)
+
+    meta_est.set_params(bar__b="something else")
+    assert meta_est.get_params()["bar__b"] == "something else"

skbase/utils/__init__.py

@@ -1,21 +1,21 @@
-#!/usr/bin/env python3 -u
-# -*- coding: utf-8 -*-
-# copyright: skbase developers, BSD-3-Clause License (see LICENSE file)
-"""Utility functionality used through `skbase`."""
-from typing import List
-
-from skbase.utils._iter import make_strings_unique
-from skbase.utils._nested_iter import flatten, is_flat, unflat_len, unflatten
-from skbase.utils._utils import subset_dict_keys
-from skbase.utils.deep_equals import deep_equals
-
-__author__: List[str] = ["RNKuhns", "fkiraly"]
-__all__: List[str] = [
-    "deep_equals",
-    "flatten",
-    "is_flat",
-    "make_strings_unique",
-    "subset_dict_keys",
-    "unflat_len",
-    "unflatten",
-]
+#!/usr/bin/env python3 -u
+# -*- coding: utf-8 -*-
+# copyright: skbase developers, BSD-3-Clause License (see LICENSE file)
+"""Utility functionality used through `skbase`."""
+from typing import List
+
+from skbase.utils._iter import make_strings_unique
+from skbase.utils._nested_iter import flatten, is_flat, unflat_len, unflatten
+from skbase.utils._utils import subset_dict_keys
+from skbase.utils.deep_equals import deep_equals
+
+__author__: List[str] = ["RNKuhns", "fkiraly"]
+__all__: List[str] = [
+    "deep_equals",
+    "flatten",
+    "is_flat",
+    "make_strings_unique",
+    "subset_dict_keys",
+    "unflat_len",
+    "unflatten",
+]