fix autopep8 and tests
@@ -37,13 +37,13 @@ class TestLogSoftmax:
return scp.special.log_softmax(a)
@testing.for_dtypes('fdFD')
- @testing.numpy_cupy_allclose(scipy_name='scp', atol=1e-8, rtol=1e-8)
+ @testing.numpy_cupy_allclose(scipy_name='scp', atol=atol, rtol=rtol)
def test_log_softmax_axis_arg(self, xp, scp, dtype):
a = xp.array([[100, 1000], [1e-10, 1e-10]])
return scp.special.log_softmax(a, axis=-1)
@testing.for_dtypes('fdFD')
- @testing.numpy_cupy_allclose(scipy_name='scp', atol=1e-5, rtol=1e-5)
+ @testing.numpy_cupy_allclose(scipy_name='scp', atol=atol, rtol=rtol)
def test_log_softmax_3d(self, xp, scp, dtype):
a = testing.shaped_random((2, 3, 4), xp, dtype)
return scp.special.log_softmax(a, axis=1)
