From 7e4e0bce9500470d5cb4a2eb5da21f9b8cc7d7d1 Mon Sep 17 00:00:00 2001
From: zilinzhu <zilinzhu@tencent.com>
Date: Tue, 18 May 2021 16:33:27 +0800
Subject: [PATCH] fix lint

---
 python/tvm/topi/random/kernel.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/python/tvm/topi/random/kernel.py b/python/tvm/topi/random/kernel.py
index de5860a9104c7..929cc99c86036 100644
--- a/python/tvm/topi/random/kernel.py
+++ b/python/tvm/topi/random/kernel.py
@@ -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