File tree Expand file tree Collapse file tree 1 file changed +7
-8
lines changed
Expand file tree Collapse file tree 1 file changed +7
-8
lines changed Original file line number Diff line number Diff line change @@ -708,14 +708,13 @@ def backfill_model_runs(
708708 .values (model_run_id = mv_run_id , model_version = mv_version )
709709 )
710710 result = local_session .get ().execute (stmt )
711- updated_rows = [ dict ( r . _mapping ) for r in result . fetchall ()]
711+ updated_count = result . rowcount or 0
712712 local_session .get ().commit ()
713713
714714 return {
715- "updated_count" : len (updated_rows ),
716- "updated_rows" : updated_rows ,
717- "latest_model_version" : {
718- "version" : mv_version ,
719- "run_id" : mv_run_id ,
720- },
721- }
715+ "inst_id" : str (inst_id ),
716+ "model_id" : str (model_id [0 ][0 ].id ),
717+ "model_name" : model_name ,
718+ "latest_model_version" : {"version" : mv_version , "run_id" : mv_run_id },
719+ "updated_count" : updated_count ,
720+ }
You can’t perform that action at this time.
0 commit comments