Fix cudf.pandas Timestamp/Timedelta not subclassing stdlib datetime objects

python/cudf/cudf/pandas/_wrappers/pandas.py

@@ -2,6 +2,7 @@
 # SPDX-License-Identifier: Apache-2.0
 import abc
 import copyreg
+import datetime
 import functools
 import importlib
 import inspect
@@ -183,6 +184,7 @@ def Timestamp_Timedelta__new__(cls, *args, **kwargs):
     "Timedelta",
     _Unusable,
     pd.Timedelta,
+    bases=(datetime.timedelta,),
     fast_to_slow=_Unusable(),
     slow_to_fast=_Unusable(),
     additional_attributes={
@@ -197,6 +199,7 @@ def Timestamp_Timedelta__new__(cls, *args, **kwargs):
     "Timestamp",
     _Unusable,
     pd.Timestamp,
+    bases=(datetime.datetime,),
     fast_to_slow=_Unusable(),
     slow_to_fast=_Unusable(),
     additional_attributes={

python/cudf/cudf/pandas/fast_slow_proxy.py

@@ -3,6 +3,7 @@
 
 from __future__ import annotations
 
+import datetime
 import functools
 import inspect
 import operator
@@ -595,6 +596,26 @@ def _fsproxy_wrap(cls, value, func):
             proxy = base_class.__new__(cls)
         elif base_class is ProxyNDarrayBase:
             proxy = base_class.__new__(cls, value)
+        elif base_class is datetime.datetime:
+            proxy = base_class.__new__(
+                cls,
+                year=value.year,
+                month=value.month,
+                day=value.day,
+                hour=value.hour,
+                minute=value.minute,
+                second=value.second,
+                microsecond=value.microsecond,
+                tzinfo=value.tzinfo,
+                fold=value.fold,
+            )
+        elif base_class is datetime.timedelta:
+            proxy = base_class.__new__(
+                cls,
+                days=value.days,
+                seconds=value.seconds,
+                microseconds=value.microseconds,
+            )
         else:
             raise TypeError(
                 f"Cannot create an proxy instance of {cls.__name__} using base class {base_class.__name__}. "
@@ -1384,6 +1405,8 @@ def is_proxy_instance(obj, type):
 
 PROXY_BASE_CLASSES: set[type] = {
     ProxyNDarrayBase,
+    datetime.datetime,
+    datetime.timedelta,
 }