Fix the timezone format for Apache Arrow (pyarrow)

This is an update of my previous patch 0b49ca6 to make the format returned by Postgres' DATE_TRUNC compatible with pyarrow (used for calling prophet).
apache · Sep 21, 2022 · c61ac76 · c61ac76
1 parent 4d12e37
commit c61ac76
Showing 1 changed file with 8 additions and 8 deletions.
diff --git a/superset/db_engine_specs/postgres.py b/superset/db_engine_specs/postgres.py
@@ -83,14 +83,14 @@ class PostgresBaseEngineSpec(BaseEngineSpec):
 
     _time_grain_expressions = {
         None: "{col}",
-        "PT1S": "DATE_TRUNC('second', {col})",
-        "PT1M": "DATE_TRUNC('minute', {col})",
-        "PT1H": "DATE_TRUNC('hour', {col})",
-        "P1D": "DATE_TRUNC('day', {col})",
-        "P1W": "DATE_TRUNC('week', {col})",
-        "P1M": "DATE_TRUNC('month', {col})",
-        "P3M": "DATE_TRUNC('quarter', {col})",
-        "P1Y": "DATE_TRUNC('year', {col})",
+        "PT1S": "DATE_TRUNC('second', {col})::timestamp",
+        "PT1M": "DATE_TRUNC('minute', {col})::timestamp",
+        "PT1H": "DATE_TRUNC('hour', {col})::timestamp",
+        "P1D": "DATE_TRUNC('day', {col})::timestamp",
+        "P1W": "DATE_TRUNC('week', {col})::timestamp",
+        "P1M": "DATE_TRUNC('month', {col})::timestamp",
+        "P3M": "DATE_TRUNC('quarter', {col})::timestamp",
+        "P1Y": "DATE_TRUNC('year', {col})::timestamp",
     }
 
     custom_errors: Dict[Pattern[str], Tuple[str, SupersetErrorType, Dict[str, Any]]] = {