fix

jorgecarleitao · Jan 8, 2024 · 4e99b57 · 4e99b57
1 parent 7751abe
commit 4e99b57
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/src/io/orc/read/mod.rs b/src/io/orc/read/mod.rs
@@ -65,7 +65,12 @@ fn infer_dt(type_: &Type, types: &[Type]) -> Result<DataType, Error> {
 }
 
 fn deserialize_validity(column: &Column, scratch: &mut Vec<u8>) -> Result<Option<Bitmap>, Error> {
-    let stream = column.get_stream(Kind::Present, std::mem::take(scratch))?;
+    let stream = match column.get_stream(Kind::Present, std::mem::take(scratch)) {
+        Ok(stream) => stream,
+        Err(_) => {
+            return Ok(None);
+        }
+    };
 
     let mut stream = decode::BooleanIter::new(stream, column.number_of_rows());
 
@@ -351,3 +356,4 @@ pub fn deserialize(data_type: DataType, column: &Column) -> Result<Box<dyn Array
         dt => Err(Error::nyi(format!("Deserializing {dt:?} from ORC"))),
     }
 }
+