Fix bigquery retry

spotify · May 3, 2023 · 5b5abfe · 5b5abfe
1 parent 6a8d898
commit 5b5abfe
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/test/contrib/bigquery_test.py b/test/contrib/bigquery_test.py
@@ -26,10 +26,7 @@
 from mock.mock import MagicMock
 
 from luigi.contrib import bigquery
-try:
-    from googleapiclient import errors
-except ImportError:
-    raise unittest.SkipTest('Unable to load googleapiclient module')
+from googleapiclient import errors
 from luigi.contrib.bigquery import BigQueryLoadTask, BigQueryTarget, BQDataset, \
     BigQueryRunQueryTask, BigQueryExtractTask
 from luigi.contrib.gcs import GCSTarget

diff --git a/tox.ini b/tox.ini
@@ -60,6 +60,7 @@ deps =
     datadog==0.22.0
     prometheus-client>=0.5.0<0.15
     dropbox: dropbox>=9.3.0<10.0
+    google-api-python-client
     jsonschema
 passenv =
     USER JAVA_HOME POSTGRES_USER DATAPROC_TEST_PROJECT_ID GCS_TEST_PROJECT_ID GCS_TEST_BUCKET GOOGLE_APPLICATION_CREDENTIALS TRAVIS_BUILD_ID TRAVIS TRAVIS_BRANCH TRAVIS_JOB_NUMBER TRAVIS_PULL_REQUEST TRAVIS_JOB_ID TRAVIS_REPO_SLUG TRAVIS_COMMIT CI DROPBOX_APP_TOKEN DOCKERHUB_TOKEN GITHUB_ACTIONS OVERRIDE_SKIP_CI_TESTS