10
10
from pandas .compat import range , string_types
11
11
from pandas .types .common import (is_numeric_v_string_like ,
12
12
is_float_dtype , is_datetime64_dtype ,
13
- is_integer_dtype , _ensure_float64 ,
14
- is_scalar ,
13
+ is_datetime64tz_dtype , is_integer_dtype ,
14
+ _ensure_float64 , is_scalar ,
15
15
_DATELIKE_DTYPES ,
16
16
needs_i8_conversion )
17
17
from pandas .types .missing import isnull
@@ -450,7 +450,7 @@ def pad_1d(values, limit=None, mask=None, dtype=None):
450
450
_method = None
451
451
if is_float_dtype (values ):
452
452
_method = getattr (algos , 'pad_inplace_%s' % dtype .name , None )
453
- elif dtype in _DATELIKE_DTYPES or is_datetime64_dtype ( values ):
453
+ elif is_datetime64_dtype ( dtype ) or is_datetime64tz_dtype ( dtype ):
454
454
_method = _pad_1d_datetime
455
455
elif is_integer_dtype (values ):
456
456
values = _ensure_float64 (values )
@@ -475,7 +475,7 @@ def backfill_1d(values, limit=None, mask=None, dtype=None):
475
475
_method = None
476
476
if is_float_dtype (values ):
477
477
_method = getattr (algos , 'backfill_inplace_%s' % dtype .name , None )
478
- elif dtype in _DATELIKE_DTYPES or is_datetime64_dtype ( values ):
478
+ elif is_datetime64_dtype ( dtype ) or is_datetime64tz_dtype ( dtype ):
479
479
_method = _backfill_1d_datetime
480
480
elif is_integer_dtype (values ):
481
481
values = _ensure_float64 (values )
@@ -501,7 +501,7 @@ def pad_2d(values, limit=None, mask=None, dtype=None):
501
501
_method = None
502
502
if is_float_dtype (values ):
503
503
_method = getattr (algos , 'pad_2d_inplace_%s' % dtype .name , None )
504
- elif dtype in _DATELIKE_DTYPES or is_datetime64_dtype ( values ):
504
+ elif is_datetime64_dtype ( dtype ) or is_datetime64tz_dtype ( dtype ):
505
505
_method = _pad_2d_datetime
506
506
elif is_integer_dtype (values ):
507
507
values = _ensure_float64 (values )
@@ -531,7 +531,7 @@ def backfill_2d(values, limit=None, mask=None, dtype=None):
531
531
_method = None
532
532
if is_float_dtype (values ):
533
533
_method = getattr (algos , 'backfill_2d_inplace_%s' % dtype .name , None )
534
- elif dtype in _DATELIKE_DTYPES or is_datetime64_dtype ( values ):
534
+ elif is_datetime64_dtype ( dtype ) or is_datetime64tz_dtype ( dtype ):
535
535
_method = _backfill_2d_datetime
536
536
elif is_integer_dtype (values ):
537
537
values = _ensure_float64 (values )
0 commit comments