Skip to content

Commit 39f33fa

Browse files
committed
fixup! fixup! refactor(pyspark): only allow timezoned timestamps with pyspark < 3.4
1 parent 084142f commit 39f33fa

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

ibis/backends/tests/test_temporal.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1025,7 +1025,7 @@ def test_timestamp_comparison_filter(backend, con, alltypes, df, func_name):
10251025
comparison_fn(alltypes.timestamp_col.cast("timestamp('UTC')"), ts)
10261026
)
10271027

1028-
if (col := df.timestamp_col).dtype.tz is None:
1028+
if getattr((col := df.timestamp_col).dtype, "tz", None) is None:
10291029
col = df.timestamp_col.dt.tz_localize("UTC")
10301030
expected = df[comparison_fn(col, ts)]
10311031
result = con.execute(expr)
@@ -1059,7 +1059,7 @@ def test_timestamp_comparison_filter_numpy(backend, con, alltypes, df, func_name
10591059

10601060
ts = pd.Timestamp(ts.item(), tz="UTC")
10611061

1062-
if (col := df.timestamp_col).dtype.tz is None:
1062+
if getattr((col := df.timestamp_col).dtype, "tz", None) is None:
10631063
col = df.timestamp_col.dt.tz_localize("UTC")
10641064
expected = df[comparison_fn(col, ts)]
10651065
result = con.execute(expr)

0 commit comments

Comments
 (0)