diff --git a/python/tvm/relay/op/strategy/generic.py b/python/tvm/relay/op/strategy/generic.py
index f9fec98373143..6c4780bc01f8e 100644
--- 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