From eb9341e674eff1ff3a17ddc8d049bcce8dc8ff1e Mon Sep 17 00:00:00 2001 From: Linus Seelinger Date: Fri, 21 Jul 2023 12:18:00 +0200 Subject: [PATCH] Fix expected output dimension in Hessian --- testing/protocol_conformity_1.0/test_model_schema.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/testing/protocol_conformity_1.0/test_model_schema.py b/testing/protocol_conformity_1.0/test_model_schema.py index a25384a..de059c0 100644 --- a/testing/protocol_conformity_1.0/test_model_schema.py +++ b/testing/protocol_conformity_1.0/test_model_schema.py @@ -495,7 +495,7 @@ def test_apply_hessian(model_url, input_value): jsonschema.validate(instance=resp.json(), schema=schema) # Check if output has dimension of output outWrt - assert len(resp.json()["output"]) == outputSizesJSON["outputSizes"][0] + assert len(resp.json()["output"]) == inputSizesJSON["inputSizes"][0] def test_apply_hessian_unsupported(model_url): resp_info = requests.get(f'{model_url}/Info')