Skip to content

Commit

Permalink
remove test_get_axis_sizes_raises_with_surplus_n (exception is a warn…
Browse files Browse the repository at this point in the history
…ing now)
  • Loading branch information
FynnBe committed Nov 12, 2024
1 parent 8504862 commit 04762bf
Showing 1 changed file with 0 additions and 13 deletions.
13 changes: 0 additions & 13 deletions tests/test_model/test_v0_5.py
Original file line number Diff line number Diff line change
Expand Up @@ -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",
Expand Down

0 comments on commit 04762bf

Please sign in to comment.