roll lint

apache · Aug 12, 2020 · 13126f1 · 13126f1
1 parent ae30ba7
commit 13126f1
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/python/tvm/relay/quantize/quantize.py b/python/tvm/relay/quantize/quantize.py
@@ -374,10 +374,9 @@ def quantize(mod, params=None, dataset=None):
 
     q_cfg = current_qconfig()
     assert q_cfg.partition_conversions in ['disabled', 'enabled', 'fully_integral']
-    if q_cfg.partition_conversions == 'disabled':
-        return mod
-    else:
+    if q_cfg.partition_conversions != 'disabled':
         quantized_dtypes = {q_cfg.dtype_input, q_cfg.dtype_weight, q_cfg.dtype_activation}
         ensure_fully_integral = q_cfg.partition_conversions == 'fully_integral'
         return partition_conversions(mod, quantized_dtypes, ensure_fully_integral)
 
+    return mod