mypy hint

py-pdf · May 22, 2022 · e36def6 · e36def6
1 parent 62afa04
commit e36def6
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/PyPDF2/_writer.py b/PyPDF2/_writer.py
@@ -1805,13 +1805,15 @@ def set_page_mode(self, mode: PagemodeType) -> None:
 
             Use :py:attr:`page_mode` instead.
         """
-        if not isinstance(mode, NameObject):
+        if isinstance(mode, NameObject):
+            mode_name: NameObject = mode
+        else:
             if mode not in self._valid_modes:
                 warnings.warn(
                     "Mode should be one of: {}".format(", ".join(self._valid_modes))
                 )
-            mode = NameObject(mode)
-        self._root_object.update({NameObject("/PageMode"): mode})
+            mode_name = NameObject(mode)
+        self._root_object.update({NameObject("/PageMode"): mode_name})
 
     def setPageMode(self, mode: PagemodeType) -> None:
         """