diff --git a/python/test/unit/language/test_random.py b/python/test/unit/language/test_random.py index 79a4e3842f86..7341e1594c18 100644 --- a/python/test/unit/language/test_random.py +++ b/python/test/unit/language/test_random.py @@ -189,7 +189,7 @@ def const_kernel(X, N, seed: tl.constexpr, dtype: tl.constexpr): const_kernel[grid](x, N, seed=seed, dtype=getattr(tl, dtype)) else: kernel[grid](x, N, seed, dtype=getattr(tl, dtype)) - assert all((x >= 0) & (x <= 1)) + assert all((x >= 0) & (x < 1)) assert scipy.stats.kstest(x.tolist(), 'uniform', args=(0, 1)).statistic < 0.01