Marshalling: Enable option orjson.OPT_SERIALIZE_NUMPY to support NumPy

amotl · amotl · commit 20e735fcf6bc · 2025-01-16T10:28:08.000+01:00
diff --git a/src/crate/client/http.py b/src/crate/client/http.py
@@ -343,7 +343,7 @@ def _create_sql_payload(stmt, args, bulk_args):
     return orjson.dumps(
         data,
         default=cratedb_json_encoder,
-        option=orjson.OPT_PASSTHROUGH_DATETIME,
+        option=(orjson.OPT_PASSTHROUGH_DATETIME | orjson.OPT_SERIALIZE_NUMPY),
     )
 
 

Original file line number	Diff line number	Diff line change
`@@ -343,7 +343,7 @@ def _create_sql_payload(stmt, args, bulk_args):`
`343`	`343`	`return orjson.dumps(`
`344`	`344`	`data,`
`345`	`345`	`default=cratedb_json_encoder,`
`346`		`- option=orjson.OPT_PASSTHROUGH_DATETIME,`
	`346`	`+ option=(orjson.OPT_PASSTHROUGH_DATETIME \| orjson.OPT_SERIALIZE_NUMPY),`
`347`	`347`	`)`
`348`	`348`
`349`	`349`