From c5c7d8c0f57be8498b45a42e21b4eb6a44487769 Mon Sep 17 00:00:00 2001 From: Michael Tiemann <72577720+MichaelTiemannOSC@users.noreply.github.com> Date: Sat, 13 Jan 2024 20:57:09 +1300 Subject: [PATCH] Explicitly declare `units` field in PintType Improve mypy compatibility by declaring `units` to be a field of PintType. Signed-off-by: Michael Tiemann <72577720+MichaelTiemannOSC@users.noreply.github.com> --- pint_pandas/pint_array.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/pint_pandas/pint_array.py b/pint_pandas/pint_array.py index 6fb96ee..ede32f0 100644 --- a/pint_pandas/pint_array.py +++ b/pint_pandas/pint_array.py @@ -2,6 +2,7 @@ import re import warnings from importlib.metadata import version +from typing import Optional import numpy as np import pandas as pd @@ -43,6 +44,7 @@ class PintType(ExtensionDtype): # str = '|O08' # base = np.dtype('O') # num = 102 + units: Optional[_Unit] = None # Filled in by `construct_from_..._string` _metadata = ("units",) _match = re.compile(r"(P|p)int\[(?P.+)\]") _cache = {}