From 63dd114a56045c1a5d886a2ac9504dbbda6c4a71 Mon Sep 17 00:00:00 2001 From: Jon Mease Date: Thu, 26 Oct 2023 14:38:50 -0400 Subject: [PATCH] cast arrow table with safe=False to allow precision truncation --- python/vegafusion/vegafusion/dataset/snowpark.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/vegafusion/vegafusion/dataset/snowpark.py b/python/vegafusion/vegafusion/dataset/snowpark.py index e20df8c8..dfe150df 100644 --- a/python/vegafusion/vegafusion/dataset/snowpark.py +++ b/python/vegafusion/vegafusion/dataset/snowpark.py @@ -92,7 +92,7 @@ def fetch_query(self, query: str, schema: pa.Schema) -> pa.Table: sp_df = self._session.sql(query) batches = [] for pd_batch in sp_df.to_pandas_batches(): - pa_tbl = to_arrow_table(pd_batch).cast(schema) + pa_tbl = to_arrow_table(pd_batch).cast(schema, safe=False) batches.extend(pa_tbl.to_batches()) return pa.Table.from_batches(batches, schema)