Update _base.py

sktime · Sep 22, 2024 · 7bf2637 · 7bf2637
1 parent 4666c65
commit 7bf2637
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/skbase/base/_base.py b/skbase/base/_base.py
@@ -1384,7 +1384,8 @@ def _clone(estimator, *, safe=True):
     found in :ref:`randomness`.
     """
     estimator_type = type(estimator)
-    # XXX: not handling dictionaries
+    if estimator_type is dict:
+        return {k: _clone(v, safe=safe) for k, v in estimator.items()}
     if estimator_type in (list, tuple, set, frozenset):
         return estimator_type([_clone(e, safe=safe) for e in estimator])
     elif not hasattr(estimator, "get_params") or isinstance(estimator, type):