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
@@ -156,7 +156,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: ...
@@ -305,7 +305,7 @@
     @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 @@
     @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 @@
     @property
     def body(self) -> Block: ...
     @property
-    def context(self) -> object: ...
+    def context(self) -> Context: ...
     @property
     def operation(self) -> Operation: ...