diff --git a/mlir/python/mlir/_mlir_libs/_mlir/ir.pyi b/mlir/python/mlir/_mlir_libs/_mlir/ir.pyi
--- a/mlir/python/mlir/_mlir_libs/_mlir/ir.pyi
+++ b/mlir/python/mlir/_mlir_libs/_mlir/ir.pyi
@@ -151,7 +151,7 @@
     @property
     def _CAPIPtr(self) -> object: ...
     @property
-    def context(self) -> object: ...
+    def context(self) -> Context: ...
 
 class Attribute:
     def __init__(self, cast_from_type: Attribute) -> None: ...
@@ -300,7 +300,7 @@
     @property
     def _CAPIPtr(self) -> object: ...
     @property
-    def context(self) -> object: ...
+    def context(self) -> Context: ...
     @property
     def is_permutation(self) -> bool: ...
     @property
@@ -694,7 +694,7 @@
     @property
     def constraints(self) -> Any: ...
     @property
-    def context(self) -> object: ...
+    def context(self) -> Context: ...
     @property
     def is_canonical_empty(self) -> bool: ...
     @property
@@ -796,7 +796,7 @@
     @property
     def body(self) -> Block: ...
     @property
-    def context(self) -> object: ...
+    def context(self) -> Context: ...
     @property
     def operation(self) -> Operation: ...