Public/nushell
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Incorporating API changes in polars 0.41

Browse Source
This commit is contained in:
Jack Wright 2024-06-26 10:49:09 -07:00
parent f3e36ef7f5
commit 83f859e1b5
1 changed files with 8 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
crates/nu_plugin_polars/src/dataframe/eager/open.rs
View File

@ -16,6 +16,7 @@ use nu_protocol::{
use std::{ use std::{
fs::File, fs::File,
io::BufReader, io::BufReader,
num::NonZeroUsize,
path::{Path, PathBuf}, path::{Path, PathBuf},
sync::Arc, sync::Arc,
}; };
@ -376,9 +377,13 @@ fn from_jsonl(
file_path: &Path, file_path: &Path,
file_span: Span, file_span: Span,
) -> Result<Value, ShellError> { ) -> Result<Value, ShellError> {
let infer_schema: usize = call let infer_schema: NonZeroUsize = call
.get_flag("infer-schema")? .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 let maybe_schema = call
.get_flag("schema")? .get_flag("schema")?
.map(|schema| NuSchema::try_from(&schema)) .map(|schema| NuSchema::try_from(&schema))
@ -541,7 +546,7 @@ fn from_csv(
.with_infer_schema_length(Some(infer_schema)) .with_infer_schema_length(Some(infer_schema))
.with_skip_rows(skip_rows.unwrap_or_default()) .with_skip_rows(skip_rows.unwrap_or_default())
.with_schema(maybe_schema.map(|s| s.into())) .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| { .map_parse_options(|options| {
options options
.with_separator( .with_separator(