address review suggestions

Signed-off-by: dchigarev <dmitry.chigarev@intel.com>

Author: dchigarev

ingress/Torch-MLIR/py_src/export_lib/utils.py

@@ -63,15 +63,13 @@ def parse_shape_str(shape: str) -> tuple[tuple[int], torch.dtype]:
     shape : str
         A string representing the shape and dtype, e.g. '1,3,224,224,float32'.
     """
-    components = shape.split(",")
-    shapes = components[:-1]
-    dtype = components[-1]
+    *shapes, dtype = shape.split(",")
     tdtype = getattr(torch, dtype)
     if tdtype is None:
         raise ValueError(f"Unsupported dtype: {dtype}")
-    if any(dim == "?" for dim in shapes):
+    if "?" in shapes:
         raise ValueError(f"Dynamic shapes are not supported yet: {shape}")
-    return (tuple(int(dim) for dim in shapes if dim), tdtype)
+    return (tuple(map(int, shapes)), tdtype)
 
 
 def generate_fake_tensor(shape: tuple[int], dtype: torch.dtype) -> torch.Tensor: