[Mlir-commits] [mlir] [mlir][python] fix up affine for (PR #74495)

[Oleksandr Alex Zinenko]

================
@@ -21,17 +22,17 @@
 
 @_ods_cext.register_operation(_Dialect, replace=True)
 class AffineForOp(AffineForOp):
-    """Specialization for the Affine for op class"""
+    """Specialization for the Affine for op class."""
 
     def __init__(
         self,
-        lower_bound,
-        upper_bound,
-        step,
-        iter_args: Optional[Union[Operation, OpView, Sequence[Value]]] = None,
+        lower_bound: Union[int, _ResultValueT, AffineMap],
+        upper_bound: Optional[Union[int, _ResultValueT, AffineMap]] = None,
+        step: Optional[Union[int, _ResultValueT]] = None,
----------------
ftynse wrote:

I don't think `step` in affine loops can be `ResultValueT`, it's an integer attribute. This should take an int or an attribute.

https://github.com/llvm/llvm-project/pull/74495