From 04762bfe5ed8d031fc7a3b65abc652781cb1286b Mon Sep 17 00:00:00 2001 From: fynnbe Date: Tue, 12 Nov 2024 13:58:56 +0100 Subject: [PATCH] remove test_get_axis_sizes_raises_with_surplus_n (exception is a warning now) --- tests/test_model/test_v0_5.py | 13 ------------- 1 file changed, 13 deletions(-) diff --git a/tests/test_model/test_v0_5.py b/tests/test_model/test_v0_5.py index 569f64fb4..678033448 100644 --- a/tests/test_model/test_v0_5.py +++ b/tests/test_model/test_v0_5.py @@ -383,19 +383,6 @@ def test_output_fixed_shape_too_small(model_data: Dict[str, Any]): assert summary.status == "failed", summary.format() -def test_get_axis_sizes_raises_with_surplus_n(model_data: Dict[str, Any]): - with ValidationContext(perform_io_checks=False): - model = ModelDescr(**model_data) - - output_tensor_id = model.inputs[0].id - output_axis_id = AxisId("y") - - with pytest.raises(ValueError): - _ = model.get_axis_sizes( - ns={(output_tensor_id, output_axis_id): 1}, batch_size=1 - ) - - def test_get_axis_sizes_raises_with_missing_n(model_data: Dict[str, Any]): model_data["inputs"][0]["axes"][2] = { "type": "space",