Update vegafusion-sql

vega · Feb 13, 2024 · 8b18f43 · 8b18f43
1 parent 8a208a5
commit 8b18f43
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/vegafusion-common/src/data/scalar.rs b/vegafusion-common/src/data/scalar.rs
@@ -2,12 +2,12 @@ use crate::error::{Result, VegaFusionError};
 use arrow::array::{new_empty_array, Array, ArrayRef, ListArray};
 use datafusion_common::DataFusionError;
 
+use arrow::datatypes::DataType;
 use datafusion_common::utils::array_into_list_array;
 pub use datafusion_common::ScalarValue;
 
 #[cfg(feature = "json")]
 use {
-    arrow::datatypes::DataType,
     serde_json::{Map, Value},
     std::ops::Deref,
     std::sync::Arc,

diff --git a/vegafusion-python-embed/src/connection.rs b/vegafusion-python-embed/src/connection.rs
@@ -96,7 +96,7 @@ impl Connection for PySqlConnection {
             let mut tables: HashMap<String, Schema> = HashMap::new();
 
             for key in tables_dict.keys() {
-                let value = tables_dict.get_item(key).unwrap();
+                let value = tables_dict.get_item(key)?.unwrap();
                 let key_string = key.extract::<String>()?;
                 let value_schema = Schema::from_pyarrow(value)?;
                 tables.insert(key_string, value_schema);

diff --git a/vegafusion-sql/src/connection/datafusion_py_datasource.rs b/vegafusion-sql/src/connection/datafusion_py_datasource.rs
@@ -148,7 +148,7 @@ impl ExecutionPlan for PyDatasourceExec {
         )?))
     }
 
-    fn statistics(&self) -> Statistics {
-        Statistics::default()
+    fn statistics(&self) -> datafusion_common::Result<Statistics> {
+        Ok(Statistics::new_unknown(self.schema().as_ref()))
     }
 }