Skip to content

Commit 9b1525a

Browse files
authored
deps: include pyarrow as a dependency (#529)
* deps: include `pyarrow` as a dependency Test with a minimum version of pyarrow to confirm compatibility. * fix unit test * fix unit tests * try pyarrow 11 * try pyarrow 8 * skip failing astype test
1 parent 893fe15 commit 9b1525a

File tree

5 files changed

+17
-7
lines changed

5 files changed

+17
-7
lines changed

noxfile.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -341,8 +341,8 @@ def run_system(
341341
pytest_cmd.extend(extra_pytest_options)
342342
session.run(
343343
*pytest_cmd,
344-
test_folder,
345344
*session.posargs,
345+
test_folder,
346346
)
347347

348348

setup.py

+1
Original file line numberDiff line numberDiff line change
@@ -47,6 +47,7 @@
4747
"ibis-framework[bigquery] >=8.0.0,<9.0.0dev",
4848
# TODO: Relax upper bound once we have fixed `system_prerelease` tests.
4949
"pandas >=1.5.0",
50+
"pyarrow >=8.0.0",
5051
"pydata-google-auth >=1.8.2",
5152
"requests >=2.27.1",
5253
"scikit-learn >=1.2.2",

testing/constraints-3.10.txt

+8-6
Original file line numberDiff line numberDiff line change
@@ -1,15 +1,17 @@
11
# Keep in sync with colab/containers/requirements.core.in image
2-
google-auth==2.17.3
2+
google-auth==2.27.0
33
ipykernel==5.5.6
44
ipython==7.34.0
5-
notebook==6.4.8
6-
pandas==1.5.3
7-
portpicker==1.3.9
8-
requests==2.27.1
9-
tornado==6.3.1
5+
notebook==6.5.5
6+
pandas==2.0.3
7+
pandas-stubs==2.0.3.230814
8+
portpicker==1.5.2
9+
requests==2.31.0
10+
tornado==6.3.3
1011
absl-py==1.4.0
1112
debugpy==1.6.6
1213
ipywidgets==7.7.1
1314
matplotlib==3.7.1
1415
psutil==5.9.5
16+
seaborn==0.13.1
1517
traitlets==5.7.1

testing/constraints-3.9.txt

+1
Original file line numberDiff line numberDiff line change
@@ -12,6 +12,7 @@ google-cloud-resource-manager==1.10.3
1212
google-cloud-storage==2.0.0
1313
ibis-framework==8.0.0
1414
pandas==1.5.0
15+
pyarrow==8.0.0
1516
pydata-google-auth==1.8.2
1617
requests==2.27.1
1718
scikit-learn==1.2.2

tests/system/small/test_series.py

+6
Original file line numberDiff line numberDiff line change
@@ -2783,6 +2783,12 @@ def test_string_astype_float():
27832783

27842784

27852785
def test_string_astype_date():
2786+
if int(pa.__version__.split(".")[0]) < 15:
2787+
pytest.skip(
2788+
"Avoid pyarrow.lib.ArrowNotImplementedError: "
2789+
"Unsupported cast from string to date32 using function cast_date32."
2790+
)
2791+
27862792
pd_series = pd.Series(["2014-08-15", "2215-08-15", "2016-02-29"]).astype(
27872793
pd.ArrowDtype(pa.string())
27882794
)

0 commit comments

Comments
 (0)