Small refactoring

elastic · May 8, 2024 · 0d8988e · 0d8988e
1 parent e38832d
commit 0d8988e
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/packages/kbn-unified-data-table/src/utils/get_render_cell_value.tsx b/packages/kbn-unified-data-table/src/utils/get_render_cell_value.tsx
@@ -66,6 +66,11 @@ export const getRenderCellValueFn = ({
     const field = dataView.fields.getByName(columnId);
     const ctx = useContext(UnifiedDataTableContext);
 
+    const onTreeExpand = useCallback(() => {
+      // expand row {rowIndex}
+      toggleRowHeight?.(rowIndex);
+    }, [rowIndex]);
+
     useEffect(() => {
       if (row?.isAnchor) {
         setCellProps({
@@ -128,12 +133,7 @@ export const getRenderCellValueFn = ({
     }
 
     if (field?.type === '_source' || useTopLevelObjectColumns) {
-      const onTreeExpand = useCallback(() => {
-        // expand row {rowIndex}
-        toggleRowHeight(rowIndex);
-      }, [rowIndex]);
-
-      const isSingleRow = autoHeightRows.includes(rowIndex) ? false : rowHeightLines === 0;
+      const isSingleRow = autoHeightRows?.includes(rowIndex) ? false : rowHeightLines === 0;
       return (
         <SourceDocument
           useTopLevelObjectColumns={useTopLevelObjectColumns}