Updates

pygmt/figure.py

@@ -544,7 +544,6 @@ def _repr_html_(self):
         wiggle,
     )
 
-
 def set_display(method=None):
     """
     Set the display method when calling :meth:`pygmt.Figure.show`.

pygmt/src/shift_origin.py

@@ -2,14 +2,11 @@
 shift_origin - Shift plot origin in x and/or y directions.
 """
 
-from __future__ import annotations
-
-from pygmt import Figure
 from pygmt.clib import Session
 from pygmt.helpers import build_arg_list
 
 
-class shift_origin(Figure):  # noqa: N801
+class shift_origin:  # noqa: N801
     r"""
     Shift plot origin in x and/or y directions.
 
@@ -61,32 +58,30 @@ class shift_origin(Figure):  # noqa: N801
     def __init__(
         self, xshift: float | str | None = None, yshift: float | str | None = None
     ):
+        """
+        Shift the plot origin in x/y directions and store the shift values.
+        """
         # self._preprocess()  # pylint: disable=protected-access
-
-        kwargs = {"T": True}
-        if xshift:
-            kwargs["X"] = xshift
-        if yshift:
-            kwargs["Y"] = yshift
+        kwdict = {"T": True, "X": xshift, "Y": yshift}
         with Session() as lib:
-            lib.call_module(module="plot", args=build_arg_list(kwargs))
-            self.saved_xshift = lib.get_common("X")  # False or xshift in inches
-            self.saved_yshift = lib.get_common("Y")  # False or yshift in inches
+            lib.call_module(module="plot", args=build_arg_list(kwdict))
+            self._xshift = lib.get_common("X")  # False or xshift in inches
+            self._yshift = lib.get_common("Y")  # False or yshift in inches
 
     def __enter__(self):
         """
-        Enter the context manager.
+        Do nothing but return self.
         """
         return self
 
     def __exit__(self, exc_type, exc_value, traceback):
         """
         Exit the context manager.
         """
-        kwargs = {"T": True}
-        if self.saved_xshift:
-            kwargs["X"] = f"{-1.0 * self.saved_xshift}i"
-        if self.saved_yshift:
-            kwargs["Y"] = f"{-1.0 * self.saved_yshift}i"
+        kwdict = {
+            "T": True,
+            "X": f"{-1.0 * self._xshift}i" if self._xshift else None,
+            "Y": f"{-1.0 * self._yshift}i" if self._yshift else None,
+        }
         with Session() as lib:
-            lib.call_module(module="plot", args=build_arg_list(kwargs))
+            lib.call_module(module="plot", args=build_arg_list(kwdict))