diff --git a/runs/frontier-fidelity-envelope-v1/test_fidelity_envelope.py b/runs/frontier-fidelity-envelope-v1/test_fidelity_envelope.py index ac0f3e0..c48ba2b 100644 --- a/runs/frontier-fidelity-envelope-v1/test_fidelity_envelope.py +++ b/runs/frontier-fidelity-envelope-v1/test_fidelity_envelope.py @@ -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["shapes"], [(512, 1), (2048, 2)]) json.dumps(module.trace_manifest_entry(parsed)) 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)