Skip to content

Commit

Permalink
Add support for pandas 2.0
Browse files Browse the repository at this point in the history
  • Loading branch information
caneff committed Sep 26, 2023
1 parent 70979bf commit c0763d2
Showing 3 changed files with 9 additions and 4 deletions.
6 changes: 3 additions & 3 deletions sdks/python/setup.py
Original file line number Diff line number Diff line change
@@ -148,7 +148,7 @@ def cythonize(*args, **kwargs):
# Exclude 1.5.0 and 1.5.1 because of
# https://github.com/pandas-dev/pandas/issues/45725
dataframe_dependency = [
'pandas>=1.4.3,!=1.5.0,!=1.5.1,<1.6;python_version>="3.8"',
'pandas>=1.4.3,!=1.5.0,!=1.5.1,<2.1;python_version>="3.8"',
]


@@ -285,13 +285,13 @@ def get_portability_package_data():
# Pinning docutils as a workaround for Sphinx issue:
# https://github.com/sphinx-doc/sphinx/issues/9727
'docutils==0.17.1',
'pandas<2.0.0',
'pandas<2.1.0',
],
'test': [
'freezegun>=0.3.12',
'joblib>=1.0.1',
'mock>=1.0.1,<6.0.0',
'pandas<2.0.0',
'pandas<2.1.0',
'parameterized>=0.7.1,<0.10.0',
'pyhamcrest>=1.9,!=1.10.0,<3.0.0',
'pyyaml>=3.12,<7.0.0',
4 changes: 4 additions & 0 deletions sdks/python/test-suites/tox/py38/build.gradle
Original file line number Diff line number Diff line change
@@ -91,6 +91,10 @@ toxTask "testPy38pandas-15", "py38-pandas-15", "${posargs}"
test.dependsOn "testPy38pandas-15"
preCommitPyCoverage.dependsOn "testPy38pandas-15"

toxTask "testPy38pandas-20", "py38-pandas-20", "${posargs}"
test.dependsOn "testPy38pandas-20"
preCommitPyCoverage.dependsOn "testPy38pandas-20"

// Create a test task for each minor version of pytorch
toxTask "testPy38pytorch-19", "py38-pytorch-19", "${posargs}"
test.dependsOn "testPy38pytorch-19"
3 changes: 2 additions & 1 deletion sdks/python/tox.ini
Original file line number Diff line number Diff line change
@@ -316,11 +316,12 @@ commands =
/bin/sh -c 'pytest -o junit_suite_name={envname} --junitxml=pytest_{envname}.xml -n 6 -m uses_pyarrow {posargs}; ret=$?; [ $ret = 5 ] && exit 0 || exit $ret'


[testenv:py{38,39,310,311}-pandas-{14,15}]
[testenv:py{38,39,310,311}-pandas-{14,15,20}]
deps =
14: pandas>=1.4.3,<1.5.0
# Exclude 1.5.0 and 1.5.1 because of https://github.com/pandas-dev/pandas/issues/45725
15: pandas>=1.5.2,<1.6.0
20: pandas>=2.0.0,<2.1.0
commands =
# Log pandas and numpy version for debugging
/bin/sh -c "pip freeze | grep -E '(pandas|numpy)'"

0 comments on commit c0763d2

Please sign in to comment.