Skip to content

Commit 06ac57e

Browse files
authored
Fix Modin .dt accessor AttributeError (#2188)
1 parent 1039e0b commit 06ac57e

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

awswrangler/athena/_read.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -65,9 +65,9 @@ def _fix_csv_types(df: pd.DataFrame, parse_dates: List[str], binaries: List[str]
6565
"""Apply data types cast to a Pandas DataFrames."""
6666
if len(df.index) > 0:
6767
for col in parse_dates:
68-
try:
68+
if pd.api.types.is_datetime64_any_dtype(df[col]):
6969
df[col] = df[col].dt.date.replace(to_replace={pd.NaT: None})
70-
except AttributeError:
70+
else:
7171
df[col] = (
7272
df[col].replace(to_replace={pd.NaT: None}).apply(lambda x: date.fromisoformat(x) if x else None)
7373
)

tests/unit/test_athena.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1401,4 +1401,4 @@ def test_athena_date_recovery(path, glue_database, glue_table):
14011401
database=glue_database,
14021402
ctas_approach=False,
14031403
)
1404-
assert df.equals(df2)
1404+
assert pandas_equals(df, df2)

0 commit comments

Comments
 (0)