Allow dictionary metric specification in ProximityTree

wildboar-foundation · Feb 22, 2024 · 1395bdf · 1395bdf
1 parent 20284fb
commit 1395bdf
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/wildboar/tree/_ptree.py b/src/wildboar/tree/_ptree.py
@@ -152,6 +152,7 @@ class ProximityTreeClassifier(BaseTreeClassifier):
         "metric": [
             StrOptions(_METRIC_NAMES, deprecated={"default"}),
             list,
+            dict,
         ],
         "metric_params": [None, dict],
         "metric_factories": [  # TODO(1.4)