File tree Expand file tree Collapse file tree 1 file changed +8
-4
lines changed
Expand file tree Collapse file tree 1 file changed +8
-4
lines changed Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments