From e2be0221b70a946ec9498db89fcbbaeecf1ca5d0 Mon Sep 17 00:00:00 2001 From: Maxime Beauchemin Date: Tue, 7 May 2019 19:16:27 -0500 Subject: [PATCH] Late import for optional lib pyhive (#7471) * Late import for optional lib pyhive * fix --- superset/sql_validators/presto_db.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/superset/sql_validators/presto_db.py b/superset/sql_validators/presto_db.py index 87c2d8efe..8f2be4389 100644 --- a/superset/sql_validators/presto_db.py +++ b/superset/sql_validators/presto_db.py @@ -26,7 +26,6 @@ from typing import ( ) from flask import g -from pyhive.exc import DatabaseError from superset import app, security_manager from superset.sql_parse import ParsedQuery @@ -77,6 +76,7 @@ class PrestoDBSQLValidator(BaseSQLValidator): # engine spec's handle_cursor implementation since we don't record # these EXPLAIN queries done in validation as proper Query objects # in the superset ORM. + from pyhive.exc import DatabaseError try: db_engine_spec.execute(cursor, sql) polled = cursor.poll()