Skip to content

Commit

Permalink
fix lint
Browse files Browse the repository at this point in the history
  • Loading branch information
zhuzilin committed May 18, 2021
1 parent dd8ec19 commit 7e4e0bc
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion python/tvm/topi/random/kernel.py
Original file line number Diff line number Diff line change
Expand Up @@ -504,7 +504,7 @@ def uniform(gen, low, high, out_shape, out_dtype):
Tensor of random numbers with shape `out_shape` and type `out_dtype`.
"""
new_gen, random_bits = threefry_generate(gen, out_shape)
assert out_dtype == "float32" or out_dtype == "float64"
assert out_dtype in ('float32', 'float64')
if out_dtype == "float32":
random_dtype = "uint32"
nbits = 32
Expand Down

0 comments on commit 7e4e0bc

Please sign in to comment.