build: update exon

Cargo.toml

@@ -9,8 +9,8 @@ name = "biobear"
 
 [dependencies]
 arrow = { version = "52.1.0", features = ["pyarrow"] }
-datafusion = "40"
-exon = { version = "0.30.0", features = ["default"] }
+datafusion = "41"
+exon = { version = "0.31.0", features = ["default"] }
 pyo3 = "0.21.2"
 tokio = { version = "1", features = ["rt"] }
 noodles = { version = "0.79", features = ["core"] }

src/bam_reader.rs

@@ -71,7 +71,11 @@ impl BamIndexedReader {
             config = config.with_batch_size(batch_size);
         }
 
-        let ctx = ExonSession::with_config_exon(config);
+        let ctx = ExonSession::with_config_exon(config).map_err(|e| {
+            PyErr::new::<pyo3::exceptions::PyRuntimeError, _>(format!(
+                "Error creating ExonSession: {e}"
+            ))
+        })?;
 
         let df = self._runtime.block_on(async {
             ctx.sql(&format!(

src/bcf_reader.rs

@@ -68,7 +68,7 @@ impl BCFIndexedReader {
             config = config.with_batch_size(batch_size);
         }
 
-        let ctx = ExonSession::with_config_exon(config);
+        let ctx = ExonSession::with_config_exon(config).map_err(BioBearError::from)?;
 
         let region = Region::from_str(region).map_err(|e| {
             io::Error::new(io::ErrorKind::Other, format!("Error parsing region: {e}"))

src/exon_reader.rs

@@ -48,7 +48,7 @@ impl ExonReader {
             config = config.with_batch_size(batch_size);
         }
 
-        let ctx = ExonSession::with_config_exon(config);
+        let ctx = ExonSession::with_config_exon(config).map_err(BioBearError::from)?;
 
         let df = rt.block_on(async {
             ctx.session

src/session_context.rs

@@ -34,19 +34,13 @@ pub struct BioBearSessionContext {
     ctx: ExonSession,
 }
 
-impl Default for BioBearSessionContext {
-    fn default() -> Self {
-        Self {
-            ctx: ExonSession::new_exon(),
-        }
-    }
-}
-
 #[pymethods]
 impl BioBearSessionContext {
     #[new]
-    fn new() -> PyResult<Self> {
-        Ok(Self::default())
+    fn try_new() -> PyResult<Self> {
+        let ctx = ExonSession::new_exon().unwrap();
+
+        Ok(Self { ctx })
     }
 
     /// Read one or more VCF files from the given path.
@@ -346,10 +340,10 @@ impl BioBearSessionContext {
 
 #[pyfunction]
 pub fn connect() -> PyResult<BioBearSessionContext> {
-    Ok(BioBearSessionContext::default())
+    BioBearSessionContext::try_new()
 }
 
 #[pyfunction]
 pub fn new_session() -> PyResult<BioBearSessionContext> {
-    Ok(BioBearSessionContext::default())
+    BioBearSessionContext::try_new()
 }

src/vcf_reader.rs

@@ -69,7 +69,7 @@ impl VCFIndexedReader {
             config = config.with_batch_size(batch_size);
         }
 
-        let ctx = ExonSession::with_config_exon(config);
+        let ctx = ExonSession::with_config_exon(config).map_err(BioBearError::from)?;
 
         let region = Region::from_str(region).map_err(|e| {
             io::Error::new(