fix: fix resetting height after re-computation

Author: Lodin

src/Tree.tsx

@@ -220,6 +220,10 @@ export const createTreeComputer = <
 
     if (typeof value === 'string' || typeof value === 'symbol') {
       id = value;
+
+      if (records[id as string]) {
+        updateRecordOnWalk(records[id as string]!, options);
+      }
     } else {
       ({id} = value);
       const record = records[id as string];

src/VariableSizeTree.tsx

@@ -132,6 +132,14 @@ export class VariableSizeTree<T extends VariableSizeNodeData> extends Tree<
     );
   }
 
+  public recomputeTree(options?: VariableSizeUpdateOptions): Promise<void> {
+    return super.recomputeTree(options).then(() => {
+      if (options?.useDefaultHeight) {
+        this.list.current?.resetAfterIndex(0, true);
+      }
+    });
+  }
+
   public render(): ReactNode {
     const {children, itemSize, rowComponent, treeWalker, ...rest} = this.props;