From 90b1c6b33cf2390b053ef01a2f4bb41c1b03ddf1 Mon Sep 17 00:00:00 2001 From: Mark Shields Date: Fri, 17 Sep 2021 15:31:14 -0700 Subject: [PATCH] [checkpoint] lint fix --- python/tvm/relay/op/annotation/annotation.py | 10 ++++------ 1 file changed, 4 insertions(+), 6 deletions(-) diff --git a/python/tvm/relay/op/annotation/annotation.py b/python/tvm/relay/op/annotation/annotation.py index e200ef7a92f9c..e7878aca4e05c 100644 --- 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):