File tree Expand file tree Collapse file tree 2 files changed +2
-2
lines changed
backends/intel_hpu/tests/unittests Expand file tree Collapse file tree 2 files changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -98,7 +98,7 @@ def test_out_name(self):
98
98
exe = base .Executor (self .place )
99
99
(result ,) = exe .run (feed = {"X" : np_x }, fetch_list = [out ])
100
100
expected = np .arctan (np_x )
101
- self .assertEqual ( result , expected )
101
+ self .assertTrue ( np . allclose ( result , expected ) )
102
102
103
103
def test_dygraph (self ):
104
104
with base .dygraph .guard (self .place ):
Original file line number Diff line number Diff line change @@ -98,7 +98,7 @@ def test_out_name(self):
98
98
exe = base .Executor (self .place )
99
99
(result ,) = exe .run (feed = {"X" : np_x }, fetch_list = [out ])
100
100
expected = np .arctan (np_x )
101
- self .assertEqual ( result , expected )
101
+ self .assertTrue ( np . allclose ( result , expected ) )
102
102
103
103
def test_dygraph (self ):
104
104
with base .dygraph .guard (self .place ):
You can’t perform that action at this time.
0 commit comments