Merge pull request #40284 from ROCmSoftwarePlatform:google_upstream_rocm_platform_fix_200608
PiperOrigin-RevId: 315492021 Change-Id: I3ff1493149b8a16786bb3a342198c50768e3795f
This commit is contained in:
commit
a866c95a1a
@ -159,7 +159,7 @@ class MathTest(test.TestCase, parameterized.TestCase):
|
|||||||
actual.shape, expected.shape,
|
actual.shape, expected.shape,
|
||||||
'Shape mismatch.\nActual: {}\nExpected: {}\n{}'.format(
|
'Shape mismatch.\nActual: {}\nExpected: {}\n{}'.format(
|
||||||
actual.shape, expected.shape, msg))
|
actual.shape, expected.shape, msg))
|
||||||
np.testing.assert_almost_equal(actual.tolist(), expected.tolist())
|
np.testing.assert_allclose(actual.tolist(), expected.tolist(), rtol=1e-6)
|
||||||
|
|
||||||
def testArgsort(self):
|
def testArgsort(self):
|
||||||
self._testUnaryOp(np_math_ops.argsort, np.argsort, 'argsort')
|
self._testUnaryOp(np_math_ops.argsort, np.argsort, 'argsort')
|
||||||
|
Loading…
Reference in New Issue
Block a user