Skip to content

Commit ee5e7fc

Browse files
committed
feat(navi/dr_transform): add filename:line to file reader error message
1 parent 2dbdfe1 commit ee5e7fc

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

navi/dr_transform/src/util.rs

+5-3
Original file line numberDiff line numberDiff line change
@@ -9,15 +9,17 @@ use std::{
99
pub fn load_batch_prediction_request_base64(file_name: &str) -> Vec<Vec<u8>> {
1010
let file = File::open(file_name).expect("could not read file");
1111
let mut result = vec![];
12-
for line in io::BufReader::new(file).lines() {
12+
for (mut line_count, line) in io::BufReader::new(file).lines().enumerate() {
13+
line_count += 1;
1314
match base64::decode(line.unwrap().trim()) {
1415
Ok(payload) => result.push(payload),
15-
Err(err) => println!("error decoding line {err}"),
16+
Err(err) => println!("error decoding line {file_name}:{line_count} - {err}"),
1617
}
1718
}
1819
println!("result len: {}", result.len());
19-
return result
20+
result
2021
}
22+
2123
pub fn save_to_npy<T: npyz::Serialize + AutoSerialize>(data: &[T], save_to: String) {
2224
let mut writer = WriteOptions::new()
2325
.default_dtype()

0 commit comments

Comments
 (0)