Skip to content
Snippets Groups Projects
Commit 19cd7d0f authored by Maxence Naud's avatar Maxence Naud Committed by Maxence Naud
Browse files

Upd squeeze python unit-tests

parent ec330a5a
No related branches found
No related tags found
1 merge request!361Move code from header to source
Pipeline #68289 passed
...@@ -184,9 +184,7 @@ class TestSqueeze(unittest.TestCase): ...@@ -184,9 +184,7 @@ class TestSqueeze(unittest.TestCase):
input_data.set_backend("cpu") input_data.set_backend("cpu")
test_squeeze_op.set_input(0, input_data) test_squeeze_op.set_input(0, input_data)
with self.assertRaises((RuntimeError, AssertionError)): self.assertFalse(test_squeeze_op.forward_dims())
test_squeeze_op.forward_dims()
test_squeeze_op.forward()
return return
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment