[AIRFLOW-1179] Fix Pandas 0.2x breaking Google BigQuery change

Closes apache#2279 from NielsZeilemaker/AIRFLOW-1179
bo01ean · May 9, 2017 · ac9ccb1 · ac9ccb1
1 parent 4284e64
commit ac9ccb1
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 1 deletion.
diff --git a/airflow/contrib/hooks/bigquery_hook.py b/airflow/contrib/hooks/bigquery_hook.py
@@ -24,7 +24,7 @@
 from apiclient.discovery import build, HttpError
 from googleapiclient import errors
 from builtins import range
-from pandas.io.gbq import GbqConnector, \
+from pandas_gbq.gbq import GbqConnector, \
     _parse_data as gbq_parse_data, \
     _check_google_client_version as gbq_check_google_client_version, \
     _test_google_api_imports as gbq_test_google_api_imports

diff --git a/scripts/ci/requirements.txt b/scripts/ci/requirements.txt
@@ -61,6 +61,7 @@ nose-ignore-docstring==0.2
 nose-parameterized
 nose-timer
 pandas
+pandas-gbq
 psutil>=4.2.0, <5.0.0
 psycopg2
 pydruid

diff --git a/setup.py b/setup.py
@@ -142,6 +142,7 @@ def check_previous():
     'google-api-python-client>=1.5.0, <1.6.0',
     'oauth2client>=2.0.2, <2.1.0',
     'PyOpenSSL',
+    'pandas-gbq'
 ]
 hdfs = ['snakebite>=2.7.8']
 webhdfs = ['hdfs[dataframe,avro,kerberos]>=2.0.4']