const columnDefs:()

in neuron_viewer/src/TransformerDebugger/cards/LogitsDisplay.tsx [134:191]
28 lines of code
2 McCabe index (conditional complexity)

  const columnDefs: (ColDef<LogitDatum, any> | ColGroupDef<LogitDatum>)[] = useMemo(() => {
    const defaultFloatColDefs: ColDef<LogitDatum, any> = {
      valueFormatter: (params: any) => formatFloat(params.value),
      resizable: true,
      width: 150,
      sortingOrder: ["desc", "asc"],
      sortable: true,
      comparator: compareWithUndefinedLast,
    };

    const rightColumns: (ColDef<LogitDatum, any> | ColGroupDef<LogitDatum>)[] = rightResponseData
      ? [
          {
            ...defaultFloatColDefs,
            field: "diffLogit",
            headerName: "Diff (left - right logit)",
            valueFormatter: (params: any) =>
              formatFloatWithZeroPoint(
                params.value,
                leftBaselineLogit - rightBaselineLogit,
                /* numDecimalPlaces= */ 3
              ),
          },
          {
            ...defaultFloatColDefs,
            field: "rightLogit",
            headerName: "Right logit",
            valueFormatter: (params: any) =>
              formatFloatWithZeroPoint(params.value, rightBaselineLogit),
          },
        ]
      : [];
    const leftColumns: (ColDef<LogitDatum, any> | ColGroupDef<LogitDatum>)[] = [
      {
        field: "token",
        headerName: "Token",
        width: 200,
        cellRenderer: (params: any) => {
          return (
            <div>
              {renderToken(params.value)}
              {params.data.label && (
                <span className="ml-2 text-xs text-gray-500 font-sans">({params.data.label})</span>
              )}
            </div>
          );
        },
      },
      {
        ...defaultFloatColDefs,
        field: "leftLogit",
        headerName: rightResponseData ? "Left logit" : "Logit",
        initialSort: "desc",
        valueFormatter: (params: any) => formatFloatWithZeroPoint(params.value, leftBaselineLogit),
      },
    ];
    return [...leftColumns, ...rightColumns];
  }, [rightResponseData, leftBaselineLogit, rightBaselineLogit]);