From 83f859e1b54e39b42cc24bf863d2f0c708738e79 Mon Sep 17 00:00:00 2001
From: Jack Wright <jack.wright@disqo.com>
Date: Wed, 26 Jun 2024 10:49:09 -0700
Subject: [PATCH] Incorporating API changes in polars 0.41

---
 crates/nu_plugin_polars/src/dataframe/eager/open.rs | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

diff --git a/crates/nu_plugin_polars/src/dataframe/eager/open.rs b/crates/nu_plugin_polars/src/dataframe/eager/open.rs
index f7ee467877..5f78e5d13d 100644
--- a/crates/nu_plugin_polars/src/dataframe/eager/open.rs
+++ b/crates/nu_plugin_polars/src/dataframe/eager/open.rs
@@ -16,6 +16,7 @@ use nu_protocol::{
 use std::{
     fs::File,
     io::BufReader,
+    num::NonZeroUsize,
     path::{Path, PathBuf},
     sync::Arc,
 };
@@ -376,9 +377,13 @@ fn from_jsonl(
     file_path: &Path,
     file_span: Span,
 ) -> Result<Value, ShellError> {
-    let infer_schema: usize = call
+    let infer_schema: NonZeroUsize = call
         .get_flag("infer-schema")?
-        .unwrap_or(DEFAULT_INFER_SCHEMA);
+        .and_then(NonZeroUsize::new)
+        .unwrap_or(
+            NonZeroUsize::new(DEFAULT_INFER_SCHEMA)
+                .expect("The default infer-schema should be non zero"),
+        );
     let maybe_schema = call
         .get_flag("schema")?
         .map(|schema| NuSchema::try_from(&schema))
@@ -541,7 +546,7 @@ fn from_csv(
             .with_infer_schema_length(Some(infer_schema))
             .with_skip_rows(skip_rows.unwrap_or_default())
             .with_schema(maybe_schema.map(|s| s.into()))
-            .with_columns(columns.map(Arc::new))
+            .with_columns(columns.map(|v| Arc::from(v.into_boxed_slice())))
             .map_parse_options(|options| {
                 options
                     .with_separator(