Change return type of ion and ioff to fix unbound variable errors with Pyright

Author: FeldrinH

lib/matplotlib/pyplot.py

@@ -554,7 +554,7 @@ def isinteractive() -> bool:
     return matplotlib.is_interactive()
 
 
-def ioff() -> ExitStack:
+def ioff() -> AbstractContextManager:
     """
     Disable interactive mode.
 
@@ -584,7 +584,7 @@ def ioff() -> ExitStack:
             # ...
 
     To enable optional usage as a context manager, this function returns a
-    `~contextlib.ExitStack` object, which is not intended to be stored or
+    context manager object, which is not intended to be stored or
     accessed by the user.
     """
     stack = ExitStack()
@@ -594,7 +594,7 @@ def ioff() -> ExitStack:
     return stack
 
 
-def ion() -> ExitStack:
+def ion() -> AbstractContextManager:
     """
     Enable interactive mode.
 
@@ -624,7 +624,7 @@ def ion() -> ExitStack:
             # ...
 
     To enable optional usage as a context manager, this function returns a
-    `~contextlib.ExitStack` object, which is not intended to be stored or
+    context manager object, which is not intended to be stored or
     accessed by the user.
     """
     stack = ExitStack()