From 1fa333e8da91807c080f2b5ddc656ce538f3f418 Mon Sep 17 00:00:00 2001 From: "Michael S. Molina" Date: Mon, 26 Aug 2024 17:12:41 -0300 Subject: [PATCH] fix: Partition calls from Jinja context --- superset/jinja_context.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/superset/jinja_context.py b/superset/jinja_context.py index 03ff43e8f4bcd..a2c87db765071 100644 --- a/superset/jinja_context.py +++ b/superset/jinja_context.py @@ -34,6 +34,7 @@ from superset.constants import LRU_CACHE_MAX_SIZE from superset.exceptions import SupersetTemplateException from superset.extensions import feature_flag_manager +from superset.sql_parse import Table from superset.utils import json from superset.utils.core import ( convert_legacy_filters_into_adhoc, @@ -619,7 +620,7 @@ def latest_partitions(self, table_name: str) -> Optional[list[str]]: table_name, schema = self._schema_table(table_name, self._schema) return cast(PrestoEngineSpec, self._database.db_engine_spec).latest_partition( - table_name, schema, self._database + database=self._database, table=Table(table_name, schema) )[1] def latest_sub_partition(self, table_name: str, **kwargs: Any) -> Any: @@ -631,7 +632,7 @@ def latest_sub_partition(self, table_name: str, **kwargs: Any) -> Any: return cast( PrestoEngineSpec, self._database.db_engine_spec ).latest_sub_partition( - table_name=table_name, schema=schema, database=self._database, **kwargs + database=self._database, table=Table(table_name, schema), **kwargs ) latest_partition = first_latest_partition