Skip to content

Commit b8f2d81

Browse files
Hoist r3 and lp deserializer imports to module level in fireworks_tracing.
Co-authored-by: Cursor <cursoragent@cursor.com>
1 parent ac3db7b commit b8f2d81

1 file changed

Lines changed: 2 additions & 4 deletions

File tree

eval_protocol/adapters/fireworks_tracing.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,8 @@
1616

1717
from eval_protocol.models import EvaluationRow, InputMetadata, ExecutionMetadata, Message
1818
from .base import BaseAdapter
19+
from .lp_deserializer import decompress_and_parse_lp
20+
from .r3_deserializer import decompress_and_parse_r3
1921
from .utils import extract_messages_from_data
2022
from ..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

Comments
 (0)