Skip to content

Commit

Permalink
Fix fmt and clippy
Browse files Browse the repository at this point in the history
  • Loading branch information
magicalne committed Mar 6, 2024
1 parent 105b9c6 commit 4de944e
Showing 1 changed file with 3 additions and 2 deletions.
5 changes: 3 additions & 2 deletions src/dataset.rs
Original file line number Diff line number Diff line change
Expand Up @@ -152,7 +152,8 @@ impl Dataset {

let (m, n) = dataframe.shape();

let label_series = &dataframe.select_series([label_col_name])?[0].cast(&DataType::Float32)?;
let label_series =
&dataframe.select_series([label_col_name])?[0].cast(&DataType::Float32)?;

if label_series.null_count() != 0 {
panic!("Cannot create a dataset with null values, encountered nulls when creating the label array")
Expand All @@ -176,7 +177,7 @@ impl Dataset {
feature_values.push(Vec::with_capacity(n));
}

for (_col_idx, series) in dataframe.get_columns().iter().enumerate() {
for series in dataframe.get_columns().iter() {
if series.null_count() != 0 {
panic!("Cannot create a dataset with null values, encountered nulls when creating the features array")
}
Expand Down

0 comments on commit 4de944e

Please sign in to comment.