File tree 1 file changed +5
-3
lines changed
1 file changed +5
-3
lines changed Original file line number Diff line number Diff line change @@ -9,15 +9,17 @@ use std::{
9
9
pub fn load_batch_prediction_request_base64 ( file_name : & str ) -> Vec < Vec < u8 > > {
10
10
let file = File :: open ( file_name) . expect ( "could not read file" ) ;
11
11
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 ;
13
14
match base64:: decode ( line. unwrap ( ) . trim ( ) ) {
14
15
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}" ) ,
16
17
}
17
18
}
18
19
println ! ( "result len: {}" , result. len( ) ) ;
19
- return result
20
+ result
20
21
}
22
+
21
23
pub fn save_to_npy < T : npyz:: Serialize + AutoSerialize > ( data : & [ T ] , save_to : String ) {
22
24
let mut writer = WriteOptions :: new ( )
23
25
. default_dtype ( )
You can’t perform that action at this time.
0 commit comments