From c34ee97954742f343aa4a49f12b38e2e99743acb Mon Sep 17 00:00:00 2001
From: quambene <contact.quambene@gmail.com>
Date: Fri, 5 Apr 2024 02:48:54 +0200
Subject: [PATCH] Fix write image

---
 src/sources/image.rs | 14 +++-----------
 1 file changed, 3 insertions(+), 11 deletions(-)

diff --git a/src/sources/image.rs b/src/sources/image.rs
index 834d261..9fd0026 100644
--- a/src/sources/image.rs
+++ b/src/sources/image.rs
@@ -51,23 +51,15 @@ pub fn write_image(
         let image = df
             .column(image_column)
             .context("Can't find column for images")?
-            .list()
+            .binary()
             .context("Can't convert series to chunked array")?
             .get(i);
 
         println!("Save query result to file: {}", target_path.display());
 
-        if let Some(image) = image {
-            let bytes = image
-                .u8()
-                .context("Can't convert series to chunked array")?
-                .into_iter()
-                .map(|byte| byte.expect("Can't convert series to bytes"))
-                .collect::<Vec<_>>();
-
+        if let Some(bytes) = image {
             let mut file = File::create(target_path).context("Unable to create file")?;
-            file.write_all(bytes.as_slice())
-                .context("Unable to write file.")?;
+            file.write_all(bytes).context("Unable to write file.")?;
         }
     }