From d38058383a752431d86293a9d6a5beeaad993107 Mon Sep 17 00:00:00 2001 From: Charlie-XIAO Date: Sat, 1 Jul 2023 13:02:11 +0800 Subject: [PATCH 1/2] API: add NaTType and NAType to pandas.api.typing --- pandas/api/typing/__init__.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/pandas/api/typing/__init__.py b/pandas/api/typing/__init__.py index 4c535bf81d3b6..9b5d2cb06b523 100644 --- a/pandas/api/typing/__init__.py +++ b/pandas/api/typing/__init__.py @@ -2,6 +2,9 @@ Public API classes that store intermediate results useful for type-hinting. """ +from pandas._libs import NaTType +from pandas._libs.missing import NAType + from pandas.core.groupby import ( DataFrameGroupBy, SeriesGroupBy, @@ -36,6 +39,8 @@ "ExponentialMovingWindow", "ExponentialMovingWindowGroupby", "JsonReader", + "NaTType", + "NAType", "PeriodIndexResamplerGroupby", "Resampler", "Rolling", From c3982a75aa3641fe904b816a212abb1bb137ee6b Mon Sep 17 00:00:00 2001 From: Charlie-XIAO Date: Sat, 1 Jul 2023 15:24:22 +0800 Subject: [PATCH 2/2] updated test --- pandas/tests/api/test_api.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/pandas/tests/api/test_api.py b/pandas/tests/api/test_api.py index ffed6a0935c8d..924a6db4b901b 100644 --- a/pandas/tests/api/test_api.py +++ b/pandas/tests/api/test_api.py @@ -259,6 +259,8 @@ class TestApi(Base): "ExponentialMovingWindow", "ExponentialMovingWindowGroupby", "JsonReader", + "NaTType", + "NAType", "PeriodIndexResamplerGroupby", "Resampler", "Rolling",