Remove python test
This commit is contained in:
parent
70ce119be4
commit
37b12148d3
@ -282,16 +282,6 @@ def test_evaluate():
|
|||||||
assert np.allclose(out_tensor.data, np.array([5, 8]).reshape(2, 1))
|
assert np.allclose(out_tensor.data, np.array([5, 8]).reshape(2, 1))
|
||||||
|
|
||||||
|
|
||||||
def test_evaluate_invalid_input_shape():
|
|
||||||
model = generate_add_model()
|
|
||||||
with pytest.raises(RuntimeError) as e:
|
|
||||||
assert model.evaluate(
|
|
||||||
[Tensor("float32", Shape([2, 1]))],
|
|
||||||
[Tensor("float32", Shape([3, 1])), Tensor("float32", Shape([3, 1]))],
|
|
||||||
)
|
|
||||||
assert "must be compatible with the partial shape: [2,1]" in str(e.value)
|
|
||||||
|
|
||||||
|
|
||||||
def test_get_batch():
|
def test_get_batch():
|
||||||
model = generate_add_model()
|
model = generate_add_model()
|
||||||
param = model.get_parameters()[0]
|
param = model.get_parameters()[0]
|
||||||
|
Loading…
Reference in New Issue
Block a user