Update glob.rs

Eventual-Inc · Nov 27, 2024 · e65c236 · e65c236
1 parent b9208ef
commit e65c236
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 5 deletions.
diff --git a/src/daft-parquet/src/python.rs b/src/daft-parquet/src/python.rs
@@ -225,7 +225,7 @@ pub mod pylib {
     #[pyfunction]
     pub fn read_parquet_schema(
         py: Python,
-        uri: String,
+        uri: &str,
         io_config: Option<IOConfig>,
         multithreaded_io: Option<bool>,
         coerce_int96_timestamp_unit: Option<PyTimeUnit>,
@@ -254,7 +254,7 @@ pub mod pylib {
                 .await
             };
 
-            let (schema, _) = runtime_handle.block_on(task)??;
+            let (schema, _) = runtime_handle.block_on_current_thread(task)?;
 
             Ok(Arc::new(schema).into())
         })

diff --git a/src/daft-scan/src/glob.rs b/src/daft-scan/src/glob.rs
@@ -87,9 +87,7 @@ async fn run_glob(
         .glob(glob_input, None, None, limit, io_stats, Some(file_format))
         .await?;
 
-    let stream = stream.map(|fm| Ok(fm?));
-
-    let stream = stream.boxed();
+    let stream = stream.map_err(|e| e.into());
 
     Ok(stream)
 }