pylint fix

apache · Jan 18, 2023 · 441626b · 441626b
1 parent 6d0e8d5
commit 441626b
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/python/tvm/relay/op/strategy/generic.py b/python/tvm/relay/op/strategy/generic.py
@@ -1863,7 +1863,9 @@ def wrap_compute_bernoulli(topi_compute):
     """Wrap bernoulli topi compute"""
 
     def _compute_bernoulli(attrs, inputs, _):
-        return list(topi_compute(inputs[0], inputs[1], attrs.out_shape, attrs.dis_dtype, attrs.out_dtype))
+        return list(
+            topi_compute(inputs[0], inputs[1], attrs.out_shape, attrs.dis_dtype, attrs.out_dtype)
+        )
 
     return _compute_bernoulli