CLN: Restore Union[int, float]

Author: bashtage

pandas/io/stata.py

@@ -762,7 +762,7 @@ class StataMissingValue:
         "float64": struct.unpack("<d", float64_base)[0],
     }
 
-    def __init__(self, value: float):
+    def __init__(self, value: Union[int, float]):
         self._value = value
         # Conversion to int to avoid hash issues on 32 bit platforms #8968
         value = int(value) if value < 2147483648 else float(value)
@@ -781,7 +781,7 @@ def string(self) -> str:
         return self._str
 
     @property
-    def value(self) -> float:
+    def value(self) -> Union[int, float]:
         """
         The binary representation of the missing value.
 
@@ -806,7 +806,7 @@ def __eq__(self, other: Any) -> bool:
         )
 
     @classmethod
-    def get_base_missing_value(cls, dtype: np.dtype) -> float:
+    def get_base_missing_value(cls, dtype: np.dtype) -> Union[int, float]:
         if dtype == np.int8:
             value = cls.BASE_MISSING_VALUES["int8"]
         elif dtype == np.int16: