Replace depreated pandas function with local alternative

traja/core.py

@@ -0,0 +1,5 @@
+import pandas as pd
+
+# Check whether pandas series is datetime or timedelta
+def is_datetime_or_timedelta_dtype(series: pd.Series) -> bool:
+    return pd.api.types.is_datetime64_dtype(series) or pd.api.types.is_timedelta64_dtype(series)

traja/plotting.py

@@ -14,15 +14,14 @@
 from matplotlib.axes import Axes
 from matplotlib.collections import PathCollection
 from matplotlib.figure import Figure
-from mpl_toolkits.mplot3d import Axes3D
 from pandas.core.dtypes.common import (
-    is_datetime_or_timedelta_dtype,
     is_datetime64_any_dtype,
     is_timedelta64_dtype,
 )
 
 import traja
 from traja.frame import TrajaDataFrame
+from traja.core import is_datetime_or_timedelta_dtype
 from traja.trajectory import coords_to_flow
 
 __all__ = [

traja/trajectory.py

@@ -6,7 +6,6 @@
 import numpy as np
 import pandas as pd
 from pandas.core.dtypes.common import (
-    is_datetime_or_timedelta_dtype,
     is_datetime64_any_dtype,
     is_timedelta64_dtype,
 )
@@ -15,6 +14,7 @@
 
 import traja
 from traja import TrajaDataFrame
+from traja.core import is_datetime_or_timedelta_dtype
 
 __all__ = [
     "_bins_to_tuple",