feat: add custom sort function for DensePartitions's sort_samples_and…

…_feature_values
ivy-llc · Dec 1, 2023 · 91fb282 · 91fb282
1 parent 319c283
commit 91fb282
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/ivy/functional/frontends/sklearn/tree/_splitter.py b/ivy/functional/frontends/sklearn/tree/_splitter.py
@@ -274,4 +274,7 @@ def node_split_best(
 
 
 def sort(feature_values, samples, n):
-    return 0, 0
+    if n == 0:
+        return
+    idx = ivy.argsort(feature_values)
+    return feature_values[idx], samples[idx]