diff --git a/.github/workflows/build.wheel.sh b/.github/workflows/build.wheel.sh index 78177066e..9753d7990 100755 --- a/.github/workflows/build.wheel.sh +++ b/.github/workflows/build.wheel.sh @@ -6,7 +6,7 @@ run_test() { entry=$1 CPYTHON_VERSION=$($entry -c 'import sys; print(str(sys.version_info[0])+str(sys.version_info[1]))') (cd wheelhouse && $entry -m pip install tensorflow_io-*-cp${CPYTHON_VERSION}-*.whl) - $entry -m pip install -q pytest pytest-benchmark boto3 fastavro avro-python3 scikit-image pandas pyarrow==2.0.0 google-cloud-pubsub==2.1.0 google-cloud-bigquery-storage==1.1.0 google-cloud-bigquery==2.3.1 google-cloud-storage==1.32.0 + $entry -m pip install -q pytest pytest-benchmark boto3 fastavro avro-python3 scikit-image pandas pyarrow==3.0.0 google-cloud-pubsub==2.1.0 google-cloud-bigquery-storage==1.1.0 google-cloud-bigquery==2.3.1 google-cloud-storage==1.32.0 (cd tests && $entry -m pytest --benchmark-disable -v --import-mode=append $(find . -type f \( -iname "test_*.py" ! \( -iname "test_*_eager.py" \) \))) (cd tests && $entry -m pytest --benchmark-disable -v --import-mode=append $(find . -type f \( -iname "test_*_eager.py" ! \( -iname "test_bigquery_eager.py" \) \))) # GRPC and test_bigquery_eager tests have to be executed separately because of https://github.com/grpc/grpc/issues/20034 diff --git a/WORKSPACE b/WORKSPACE index d9fb5e3d4..ddd3a6300 100644 --- a/WORKSPACE +++ b/WORKSPACE @@ -267,11 +267,11 @@ http_archive( http_archive( name = "arrow", build_file = "//third_party:arrow.BUILD", - sha256 = "ea299df9cf440cfc43393ce12ee6d9a4c9d0dfa9fde33c3bc9b70ec25520a844", - strip_prefix = "arrow-apache-arrow-2.0.0", + sha256 = "fc461c4f0a60e7470a7c58b28e9344aa8fb0be5cc982e9658970217e084c3a82", + strip_prefix = "arrow-apache-arrow-3.0.0", urls = [ - "https://storage.googleapis.com/mirror.tensorflow.org/github.com/apache/arrow/archive/apache-arrow-2.0.0.tar.gz", - "https://github.com/apache/arrow/archive/apache-arrow-2.0.0.tar.gz", + "https://storage.googleapis.com/mirror.tensorflow.org/github.com/apache/arrow/archive/apache-arrow-3.0.0.tar.gz", + "https://github.com/apache/arrow/archive/apache-arrow-3.0.0.tar.gz", ], ) diff --git a/third_party/arrow.BUILD b/third_party/arrow.BUILD index fff11cd50..85a5a5969 100644 --- a/third_party/arrow.BUILD +++ b/third_party/arrow.BUILD @@ -31,9 +31,10 @@ genrule( srcs = ["cpp/src/arrow/util/config.h.cmake"], outs = ["cpp/src/arrow/util/config.h"], cmd = ("sed " + - "-e 's/@ARROW_VERSION_MAJOR@/2/g' " + + "-e 's/@ARROW_VERSION_MAJOR@/3/g' " + "-e 's/@ARROW_VERSION_MINOR@/0/g' " + "-e 's/@ARROW_VERSION_PATCH@/0/g' " + + "-e 's/cmakedefine ARROW_USE_NATIVE_INT128/undef ARROW_USE_NATIVE_INT128/g' " + "-e 's/cmakedefine/define/g' " + "$< >$@"), )