mypy fixup

pandas-dev · jreback · Jun 7, 2021 · May 20, 2021 · May 20, 2021 · May 20, 2021
commit 069ae4c0e5a8e8abc1bdcaf6c84ec8df9bac94e7
diff --git a/pandas/_libs/tslibs/timestamps.pyi b/pandas/_libs/tslibs/timestamps.pyi
@@ -18,6 +18,7 @@ from typing import (
 import numpy as np
 
 from pandas._libs.tslibs import (
+    BaseOffset,
     NaT,
     NaTType,
     Period,
@@ -57,6 +58,8 @@ class Timestamp(datetime):
         fold: int | None= ...,
     ) -> _S | NaTType: ...
 
+    def _set_freq(self, freq: BaseOffset | None) -> None: ...
+
     @property
     def year(self) -> int: ...
     @property

diff --git a/pandas/core/arrays/datetimes.py b/pandas/core/arrays/datetimes.py
@@ -511,7 +511,9 @@ def _check_compatible_with(self, other, setitem: bool = False):
 
     def _box_func(self, x) -> Timestamp | NaTType:
         ts = Timestamp(x, tz=self.tz)
-        if ts is not NaT:
+        # Non-overlapping identity check (left operand type: "Timestamp",
+        # right operand type: "NaTType")
+        if ts is not NaT:  # type: ignore[comparison-overlap]
             # GH#41586
             # do this instead of passing to the constructor to avoid FutureWarning
             ts._set_freq(self.freq)