1616
1717from eval_protocol .models import EvaluationRow , InputMetadata , ExecutionMetadata , Message
1818from .base import BaseAdapter
19+ from .lp_deserializer import decompress_and_parse_lp
20+ from .r3_deserializer import decompress_and_parse_r3
1921from .utils import extract_messages_from_data
2022from ..common_utils import get_user_agent
2123
@@ -106,8 +108,6 @@ def convert_trace_dict_to_evaluation_row(
106108 router_replay = payloads .get ("router_replay" )
107109 if isinstance (router_replay , dict ) and router_replay .get ("data" ):
108110 try :
109- from .r3_deserializer import decompress_and_parse_r3
110-
111111 matrices , r3_meta = decompress_and_parse_r3 (router_replay ["data" ])
112112 if execution_metadata .extra is None :
113113 execution_metadata .extra = {}
@@ -119,8 +119,6 @@ def convert_trace_dict_to_evaluation_row(
119119 logprobs_payload = payloads .get ("logprobs" )
120120 if isinstance (logprobs_payload , dict ) and logprobs_payload .get ("data" ):
121121 try :
122- from .lp_deserializer import decompress_and_parse_lp
123-
124122 logprobs , token_ids , lp_meta = decompress_and_parse_lp (logprobs_payload ["data" ])
125123 if execution_metadata .extra is None :
126124 execution_metadata .extra = {}
0 commit comments