From e65c236b8a6c2b6329fe00293d1cd0ef6d72cc31 Mon Sep 17 00:00:00 2001 From: Andrew Gazelka Date: Tue, 26 Nov 2024 21:23:09 -0800 Subject: [PATCH] Update glob.rs --- src/daft-parquet/src/python.rs | 4 ++-- src/daft-scan/src/glob.rs | 4 +--- 2 files changed, 3 insertions(+), 5 deletions(-) diff --git a/src/daft-parquet/src/python.rs b/src/daft-parquet/src/python.rs index 4a1ec4ff97..78dbd29574 100644 --- 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, multithreaded_io: Option, coerce_int96_timestamp_unit: Option, @@ -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 index 5fcf11a2d5..899e0ebc89 100644 --- 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) }