Fixes initialDepth regression (#103, fixes #99)

bkrem · May 10, 2018 · e7914fc · e7914fc
1 parent c8a9e3a
commit e7914fc
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/Tree/index.js b/src/Tree/index.js
@@ -313,12 +313,12 @@ export default class Tree extends React.Component {
       .children(d => (d._collapsed ? null : d._children));
 
     const rootNode = this.state.data[0];
-    const nodes = tree.nodes(rootNode);
-    const links = tree.links(nodes);
+    let nodes = tree.nodes(rootNode);
 
     // set `initialDepth` on first render if specified
     if (initialDepth !== undefined && this.internalState.initialRender) {
       this.setInitialTreeDepth(nodes, initialDepth);
+      nodes = tree.nodes(rootNode);
     }
 
     if (depthFactor) {
@@ -327,6 +327,7 @@ export default class Tree extends React.Component {
       });
     }
 
+    const links = tree.links(nodes);
     return { nodes, links };
   }