accept boolean type in dataframe column categories

briefercloud · Jan 7, 2025 · 7972b55 · 7972b55
1 parent 784fd8b
commit 7972b55
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/packages/types/src/index.ts b/packages/types/src/index.ts
@@ -209,7 +209,7 @@ export type DataFrameNumberColumn = z.infer<typeof DataFrameNumberColumn>
 export const DataFrameStringColumn = z.object({
   name: z.union([z.string(), z.number()]),
   type: NumpyStringTypes.or(NumpyJsonTypes),
-  categories: z.array(z.string().or(z.number())).optional(),
+  categories: z.array(z.string().or(z.number()).or(z.boolean())).optional(),
 })
 export type DataFrameStringColumn = z.infer<typeof DataFrameStringColumn>