Merge pull request #70 from GoogleCloudPlatform/issue69-impala-connect

use ibis.impala.connect function
GoogleCloudPlatform · Aug 19, 2020 · 569a96a · 569a96a
2 parents af78ba3 + 5f80e89
commit 569a96a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/data_validation/data_validation.py b/data_validation/data_validation.py
@@ -18,7 +18,7 @@
 
 import google.oauth2.service_account
 from ibis.bigquery.client import BigQueryClient
-from ibis.impala.client import ImpalaClient
+import ibis.impala
 import ibis.pandas
 from ibis.sql.mysql.client import MySQLClient
 from ibis.sql.postgres.client import PostgreSQLClient
@@ -42,7 +42,7 @@
 
 CLIENT_LOOKUP = {
     "BigQuery": BigQueryClient,
-    "Impala": ImpalaClient,
+    "Impala": ibis.impala.connect,
     "MySQL": MySQLClient,
     "Postgres": PostgreSQLClient,
     "Teradata": TeradataClient,