diff options
Diffstat (limited to 'sqlglot/dialects/spark.py')
-rw-r--r-- | sqlglot/dialects/spark.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/sqlglot/dialects/spark.py b/sqlglot/dialects/spark.py index 939f2fd..b7d1641 100644 --- a/sqlglot/dialects/spark.py +++ b/sqlglot/dialects/spark.py @@ -7,10 +7,10 @@ from sqlglot.dialects.spark2 import Spark2 from sqlglot.helper import seq_get -def _parse_datediff(args: t.Sequence) -> exp.Expression: +def _parse_datediff(args: t.List) -> exp.Expression: """ Although Spark docs don't mention the "unit" argument, Spark3 added support for - it at some point. Databricks also supports this variation (see below). + it at some point. Databricks also supports this variant (see below). For example, in spark-sql (v3.3.1): - SELECT DATEDIFF('2020-01-01', '2020-01-05') results in -4 @@ -36,7 +36,7 @@ def _parse_datediff(args: t.Sequence) -> exp.Expression: class Spark(Spark2): class Parser(Spark2.Parser): FUNCTIONS = { - **Spark2.Parser.FUNCTIONS, # type: ignore + **Spark2.Parser.FUNCTIONS, "DATEDIFF": _parse_datediff, } |