about summary refs log tree commit diff
diff options
context:
space:
mode:
authorPhillip Cloud <417981+cpcloud@users.noreply.github.com>2022-05-07 13:13:35 -0400
committerPhillip Cloud <417981+cpcloud@users.noreply.github.com>2022-05-10 09:55:07 -0400
commit15751e9f727b7557ace90a03d52cfb0464dbdc12 (patch)
tree317f628655949a7727c5da71df174bfe79a0b4ed
parent3bf217495b166a0513788c16aa96efba5425d6bc (diff)
python3Packages.google-cloud-bigquery: patch for arrow 8 and parallelize tests
-rw-r--r--pkgs/development/python-modules/google-cloud-bigquery/default.nix7
1 files changed, 7 insertions, 0 deletions
diff --git a/pkgs/development/python-modules/google-cloud-bigquery/default.nix b/pkgs/development/python-modules/google-cloud-bigquery/default.nix
index 3a5142a9ceffe..a98bac95963d6 100644
--- a/pkgs/development/python-modules/google-cloud-bigquery/default.nix
+++ b/pkgs/development/python-modules/google-cloud-bigquery/default.nix
@@ -16,6 +16,7 @@
 , proto-plus
 , psutil
 , pyarrow
+, pytest-xdist
 }:
 
 buildPythonPackage rec {
@@ -28,6 +29,11 @@ buildPythonPackage rec {
     sha256 = "sha256-UmW6BEV44Ucdg/hUGSQk/kyDnB+Hsyx4q3AXTQe89hI=";
   };
 
+  postPatch = ''
+    substituteInPlace setup.py \
+      --replace 'pyarrow >= 3.0.0, < 8.0dev' 'pyarrow >= 3.0.0, < 9.0dev'
+  '';
+
   propagatedBuildInputs = [
     google-cloud-core
     google-cloud-bigquery-storage
@@ -47,6 +53,7 @@ buildPythonPackage rec {
     google-cloud-datacatalog
     google-cloud-storage
     pytestCheckHook
+    pytest-xdist
   ];
 
   # prevent google directory from shadowing google imports