gahow/aituner
0
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update prefix trace parser test contract

Browse Source
This commit is contained in:
Gahow Wang
2026-07-19 15:35:17 +08:00
parent 9c34650746
commit f4a75aa8e4
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
runs/frontier-fidelity-envelope-v1/test_fidelity_envelope.py
View File

@@ -150,12 +150,14 @@ class FidelityEnvelopeTest(unittest.TestCase):
}, },
] ]
) )
parsed = module.parse_trace(f"u0p01={trace}") parsed = module.parse_trace(f"u0p01={trace}", prefix_caching=False)
self.assertEqual(parsed["requests"], 2) self.assertEqual(parsed["requests"], 2)
self.assertEqual(parsed["shapes"], [(512, 1), (2048, 2)]) self.assertEqual(parsed["shapes"], [(512, 1), (2048, 2)])
json.dumps(module.trace_manifest_entry(parsed)) json.dumps(module.trace_manifest_entry(parsed))
uniform = module.parse_trace( uniform = module.parse_trace(
f"r1={trace}", rate_contract="uniform-spacing" f"r1={trace}",
rate_contract="uniform-spacing",
prefix_caching=False,
) )
self.assertEqual(uniform["offered_request_rate"], 1.0) self.assertEqual(uniform["offered_request_rate"], 1.0)