[checkpoint] lint fix

apache · Sep 17, 2021 · 90b1c6b · 90b1c6b
1 parent d9c9038
commit 90b1c6b
Showing 1 changed file with 4 additions and 6 deletions.
diff --git a/python/tvm/relay/op/annotation/annotation.py b/python/tvm/relay/op/annotation/annotation.py
@@ -25,13 +25,11 @@
 def _device_to_int(device):
     if isinstance(device, _Device):
         return device.device_type
-    elif isinstance(device, str):
+    if isinstance(device, str):
         return _nd.device(device).device_type
-    else:
-        raise ValueError(
-            "device is expected to be the type of Device or "
-            "str, but received %s" % (type(device))
-        )
+    raise ValueError(
+        "device is expected to be the type of Device or " "str, but received %s" % (type(device))
+    )
 
 
 def on_device(data, device, is_fixed=False):