corrected position of dataframes (#3776)

This commit is contained in:
Fernando Herrera 2021-07-14 14:46:32 +01:00 committed by GitHub
parent 279329bfaa
commit bb218b824e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -104,6 +104,7 @@ fn command(mut args: CommandArgs) -> Result<OutputStream, ShellError> {
// Checking the column types before performing the join // Checking the column types before performing the join
check_column_datatypes( check_column_datatypes(
df.as_ref(), df.as_ref(),
r_df.as_ref(),
&l_col_string, &l_col_string,
&l_col_span, &l_col_span,
&r_col_string, &r_col_string,
@ -125,7 +126,8 @@ fn command(mut args: CommandArgs) -> Result<OutputStream, ShellError> {
} }
fn check_column_datatypes<T: AsRef<str>>( fn check_column_datatypes<T: AsRef<str>>(
df: &polars::prelude::DataFrame, df_l: &polars::prelude::DataFrame,
df_r: &polars::prelude::DataFrame,
l_cols: &[T], l_cols: &[T],
l_col_span: &Span, l_col_span: &Span,
r_cols: &[T], r_cols: &[T],
@ -146,11 +148,11 @@ fn check_column_datatypes<T: AsRef<str>>(
} }
for (l, r) in l_cols.iter().zip(r_cols.iter()) { for (l, r) in l_cols.iter().zip(r_cols.iter()) {
let l_series = df let l_series = df_l
.column(l.as_ref()) .column(l.as_ref())
.map_err(|e| parse_polars_error::<&str>(&e, &l_col_span, None))?; .map_err(|e| parse_polars_error::<&str>(&e, &l_col_span, None))?;
let r_series = df let r_series = df_r
.column(r.as_ref()) .column(r.as_ref())
.map_err(|e| parse_polars_error::<&str>(&e, &r_col_span, None))?; .map_err(|e| parse_polars_error::<&str>(&e, &r_col_span, None))?;