fix sanity

apache · Jan 15, 2021 · a59e579 · a59e579
1 parent e4bea6c
commit a59e579
Showing 1 changed file with 5 additions and 9 deletions.
diff --git a/python/mxnet/contrib/onnx/mx2onnx/_op_translations.py b/python/mxnet/contrib/onnx/mx2onnx/_op_translations.py
@@ -3014,7 +3014,7 @@ def convert_reshape_like(node, **kwargs):
         if lhs_end >= 0:
             nodes += [
                 create_tensor([lhs_end], name+'_lhs_end', kwargs["initializer"]),
-            ]        
+            ]
         else:
             nodes += [
                 create_tensor([lhs_end], name+'_lhs_end_neg', kwargs["initializer"]),
@@ -3030,24 +3030,20 @@ def convert_reshape_like(node, **kwargs):
         if rhs_end >= 0:
             nodes += [
                 create_tensor([rhs_end], name+'_rhs_end', kwargs["initializer"]),
-            ]        
+            ]
         else:
             nodes += [
                 create_tensor([rhs_end], name+'_rhs_end_neg', kwargs["initializer"]),
                 make_node('Add', [name+'_rhs_dim', name+'_rhs_end_neg'], [name+'_rhs_end']),
             ]
 
     nodes += [
-        make_node('Slice', [name+'_lhs_shape', name+'_0', name+'_lhs_begin'],
-                      [name+'_slice0_out']),
-        make_node('Slice', [name+'_rhs_shape', name+'_rhs_begin', name+'_rhs_end'],
-                      [name+'_slice1_out']),
+        make_node('Slice', [name+'_lhs_shape', name+'_0', name+'_lhs_begin'], [name+'_slice0_out']),
+        make_node('Slice', [name+'_rhs_shape', name+'_rhs_begin', name+'_rhs_end'], [name+'_slice1_out']),
         make_node('Concat', [name+'_slice0_out', name+'_slice1_out'], [name+'_concat0_out'], axis=0),
-        make_node('Slice', [name+'_lhs_shape', name+'_lhs_end', name+'_lhs_dim'],
-                      [name+'_slice2_out']),
+        make_node('Slice', [name+'_lhs_shape', name+'_lhs_end', name+'_lhs_dim'], [name+'_slice2_out']),
         make_node('Concat', [name+'_concat0_out', name+'_slice2_out'], [name+'_concat1_out'], axis=0),
         make_node('Reshape', [lhs, name+'_concat1_out'], [name], name=name)
     ]
 
     return nodes
-