Fix expected output dimension in Hessian

UM-Bridge · Jul 21, 2023 · eb9341e · eb9341e
1 parent 4374ea1
commit eb9341e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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')