[Mlir-commits] [mlir] e1fdd80 - Fixed the type of context in type stubs for MLIR Python bindings

[Alex Zinenko]

Author: Sergei Lebedev
Date: 2022-03-31T14:28:10+02:00
New Revision: e1fdd8048cf407b3d16a48282a5e1dbf5ba4a32e

URL: https://github.com/llvm/llvm-project/commit/e1fdd8048cf407b3d16a48282a5e1dbf5ba4a32e
DIFF: https://github.com/llvm/llvm-project/commit/e1fdd8048cf407b3d16a48282a5e1dbf5ba4a32e.diff

LOG: Fixed the type of context in type stubs for MLIR Python bindings

Reviewed By: ftynse

Differential Revision: https://reviews.llvm.org/D122795

Added: 
    

Modified: 
    mlir/python/mlir/_mlir_libs/_mlir/ir.pyi

Removed: 
    


################################################################################
diff  --git a/mlir/python/mlir/_mlir_libs/_mlir/ir.pyi b/mlir/python/mlir/_mlir_libs/_mlir/ir.pyi
index 7b1667fa50214..20d9c919c249f 100644
--- a/mlir/python/mlir/_mlir_libs/_mlir/ir.pyi
+++ b/mlir/python/mlir/_mlir_libs/_mlir/ir.pyi
@@ -156,7 +156,7 @@ class AffineExpr:
     @property
     def _CAPIPtr(self) -> object: ...
     @property
-    def context(self) -> object: ...
+    def context(self) -> Context: ...
 
 class Attribute:
     def __init__(self, cast_from_type: Attribute) -> None: ...
@@ -305,7 +305,7 @@ class AffineMap:
     @property
     def _CAPIPtr(self) -> object: ...
     @property
-    def context(self) -> object: ...
+    def context(self) -> Context: ...
     @property
     def is_permutation(self) -> bool: ...
     @property
@@ -699,7 +699,7 @@ class IntegerSet:
     @property
     def constraints(self) -> Any: ...
     @property
-    def context(self) -> object: ...
+    def context(self) -> Context: ...
     @property
     def is_canonical_empty(self) -> bool: ...
     @property
@@ -802,7 +802,7 @@ class Module:
     @property
     def body(self) -> Block: ...
     @property
-    def context(self) -> object: ...
+    def context(self) -> Context: ...
     @property
     def operation(self) -> Operation: ...