format

Author: ManfeiBai

torch_xla/experimental/fori_loop.py

@@ -18,6 +18,7 @@ def while_loop(cond_fn, body_fn, operands):
   # operands: (Tuple of possibly nested dict/list/tuple of tensors)
   return _xla_while_loop(cond_fn, body_fn, operands)
 
+
 def _xla_while_loop(cond_fn, body_fn, operands):
 
   def op_fn(internal_x):
@@ -33,7 +34,7 @@ def body_fn_placeholder(counter, internal_x):
       return xb.Op.tuple((next_counter, internal_x))
 
     zero = xb.Op.scalar(internal_x.builder(), 0, dtype=xb.Type.S32)
-    w = xb.Op.mkwhile((zero, internal_x), cond_fn_placeholder, 
+    w = xb.Op.mkwhile((zero, internal_x), cond_fn_placeholder,
                       body_fn_placeholder)
     return w.get_tuple_element(1)