File tree Expand file tree Collapse file tree 1 file changed +5
-3
lines changed Expand file tree Collapse file tree 1 file changed +5
-3
lines changed Original file line number Diff line number Diff line change @@ -9,15 +9,17 @@ use std::{
99pub 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+
2123pub fn save_to_npy < T : npyz:: Serialize + AutoSerialize > ( data : & [ T ] , save_to : String ) {
2224 let mut writer = WriteOptions :: new ( )
2325 . default_dtype ( )
You can’t perform that action at this time.
0 commit comments