diff --git a/splitgraph/core/sql/__init__.py b/splitgraph/core/sql/__init__.py index 7a1523cf..9ab438de 100644 --- a/splitgraph/core/sql/__init__.py +++ b/splitgraph/core/sql/__init__.py @@ -3,7 +3,6 @@ import re from typing import Callable, Dict, List, Optional, Sequence, Tuple, Union -import pglast.node from psycopg2.sql import SQL, Composed, Identifier from splitgraph.config import SPLITGRAPH_META_SCHEMA from splitgraph.core.sql._validation import ( @@ -14,6 +13,7 @@ try: from pglast import parse_sql + import pglast.node from pglast.node import Node, Scalar from pglast.parser import ParseError from pglast.stream import IndentedStream