Skip to content

Commit d18aa78

Browse files
committed
lint
1 parent 6bb4907 commit d18aa78

File tree

1 file changed

+8
-4
lines changed

1 file changed

+8
-4
lines changed

exir/emit/_emitter.py

Lines changed: 8 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -981,9 +981,11 @@ def _emit_scan(
981981
"""
982982
combine_fn, init, xs, additional_inputs = args
983983

984-
assert isinstance(
985-
subemitter_binding_output_values, (list, tuple)
986-
), f"Expected list for subemitter_binding_output_values. Got {type(subemitter_binding_output_values).__name__}: {subemitter_binding_output_values}."
984+
assert isinstance(subemitter_binding_output_values, (list, tuple)), (
985+
f"Expected list for subemitter_binding_output_values. "
986+
f"Got {type(subemitter_binding_output_values).__name__}: "
987+
f"{subemitter_binding_output_values}."
988+
)
987989

988990
assert isinstance(combine_fn, torch.fx.GraphModule)
989991
assert isinstance(init, (list, tuple))
@@ -997,7 +999,9 @@ def _emit_scan(
997999
y_outputs = list(subemitter_binding_output_values[num_carry:])
9981000

9991001
if num_xs < 1:
1000-
raise RuntimeError(f"Scan requires at least one xs tensor to scan over but got {num_xs}")
1002+
raise RuntimeError(
1003+
f"Scan requires at least one xs tensor to scan over but got {num_xs}"
1004+
)
10011005

10021006
iter_idx = self._emit_evalue(EValue(Int(0)))
10031007

0 commit comments

Comments
 (0)