Skip to content

Commit c376149

Browse files
Backport PR #56063 on branch 2.1.x (CI: Fix pyarrow nightly ci failure) (#56069)
Backport PR #56063: CI: Fix pyarrow nightly ci failure Co-authored-by: Patrick Hoefler <[email protected]>
1 parent 19bbe06 commit c376149

File tree

2 files changed

+8
-1
lines changed

2 files changed

+8
-1
lines changed

pandas/compat/pyarrow.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,7 @@
1717
pa_version_under13p0 = _palv < Version("13.0.0")
1818
pa_version_under14p0 = _palv < Version("14.0.0")
1919
pa_version_under14p1 = _palv < Version("14.0.1")
20+
pa_version_under15p0 = _palv < Version("15.0.0")
2021
except ImportError:
2122
pa_version_under7p0 = True
2223
pa_version_under8p0 = True
@@ -27,3 +28,4 @@
2728
pa_version_under13p0 = True
2829
pa_version_under14p0 = True
2930
pa_version_under14p1 = True
31+
pa_version_under15p0 = True

pandas/tests/io/test_parquet.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,7 @@
1919
pa_version_under8p0,
2020
pa_version_under11p0,
2121
pa_version_under13p0,
22+
pa_version_under15p0,
2223
)
2324

2425
import pandas as pd
@@ -755,7 +756,10 @@ def test_unsupported_float16(self, pa):
755756
# Not able to write float 16 column using pyarrow.
756757
data = np.arange(2, 10, dtype=np.float16)
757758
df = pd.DataFrame(data=data, columns=["fp16"])
758-
self.check_external_error_on_write(df, pa, pyarrow.ArrowException)
759+
if pa_version_under15p0:
760+
self.check_external_error_on_write(df, pa, pyarrow.ArrowException)
761+
else:
762+
check_round_trip(df, pa)
759763

760764
@pytest.mark.xfail(
761765
is_platform_windows(),
@@ -764,6 +768,7 @@ def test_unsupported_float16(self, pa):
764768
"dtypes are passed to_parquet function in windows"
765769
),
766770
)
771+
@pytest.mark.skipif(not pa_version_under15p0, reason="float16 works on 15")
767772
@pytest.mark.parametrize("path_type", [str, pathlib.Path])
768773
def test_unsupported_float16_cleanup(self, pa, path_type):
769774
# #44847, #44914

0 commit comments

Comments
 (0)