fix(sankey): fix broken custom node sorting (#498)

plouc · Mar 28, 2019 · 319e7e0 · 319e7e0
1 parent 41dd564
commit 319e7e0
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/packages/sankey/src/enhance.js b/packages/sankey/src/enhance.js
@@ -45,8 +45,10 @@ export default Component =>
             getLabel: getLabelGenerator(label, labelFormat),
         })),
         withPropsOnChange(['sort'], ({ sort }) => {
-            let sortFunction
-            if (sort === 'input') {
+            let sortFunction = sort
+            if (sort === 'auto') {
+                sortFunction = undefined
+            } else if (sort === 'input') {
                 sortFunction = null
             } else if (sort === 'ascending') {
                 sortFunction = (a, b) => a.value - b.value