fix(table): allow string literals for showColumnSummaries (#3307)

marimo-team · Dec 29, 2024 · 0f639ad · 0f639ad
1 parent fda938f
commit 0f639ad
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/frontend/src/plugins/impl/DataTablePlugin.tsx b/frontend/src/plugins/impl/DataTablePlugin.tsx
@@ -69,7 +69,7 @@ interface Data<T> {
   selection: "single" | "multi" | null;
   showDownload: boolean;
   showFilters: boolean;
-  showColumnSummaries: boolean;
+  showColumnSummaries: boolean | "stats" | "chart";
   rowHeaders: string[];
   fieldTypes?: FieldTypesWithExternalType | null;
   freezeColumnsLeft?: string[];
@@ -112,7 +112,9 @@ export const DataTablePlugin = createPlugin<S>("marimo-table")
       selection: z.enum(["single", "multi"]).nullable().default(null),
       showDownload: z.boolean().default(false),
       showFilters: z.boolean().default(false),
-      showColumnSummaries: z.boolean().default(true),
+      showColumnSummaries: z
+        .union([z.boolean(), z.enum(["stats", "chart"])])
+        .default(true),
       rowHeaders: z.array(z.string()),
       freezeColumnsLeft: z.array(z.string()).optional(),
       freezeColumnsRight: z.array(z.string()).optional(),