Fix pylint

python/tvm/relay/frontend/pytorch.py

@@ -304,8 +304,7 @@ def _impl(inputs, input_types):
             if isinstance(dim, tvm.tir.Any):
                 end = _op.shape_of(data)
                 break
-            else:
-                end.append(int(dim))
+            end.append(int(dim))
 
         begin = [0] * ndim
         dim = int(inputs[1])
@@ -355,8 +354,8 @@ def _impl(inputs, input_types):
                 end[dim] = target_end
             else:
                 all_static = True
-                for i in range(len(dshape)):
-                    if i != dim and isinstance(dshape[i], tvm.tir.Any):
+                for i, shape_dim in enumerate(dshape):
+                    if i != dim and isinstance(shape_dim, tvm.tir.Any):
                         all_static = False
 
                 if all_static:
@@ -1132,7 +1131,7 @@ def _impl(inputs, input_types):
             new_shape.append(0)
         out = _op.reshape(data, new_shape)
         if squeeze_axes:
-            out =  _op.squeeze(out, axis=squeeze_axes)
+            out = _op.squeeze(out, axis=squeeze_axes)
         return out
     return _impl
 
@@ -2942,7 +2941,8 @@ def get_var(name, val):
     # of the eariler loop into loop variables of the next loop.
     # So the variable corresponding to the first loop output appears free in the second loop body.
     free_vars = [var for var in _get_free_vars_from_block(body_block)
-                 if var in outputs and not isinstance(outputs[var], (_expr.Constant, int, float, str))
+                 if var in outputs and
+                 not isinstance(outputs[var], (_expr.Constant, int, float, str))
                  and outputs[var]]
 
     prev_outputs = {}