From 2a7d92a0eb9025d43f1e09307c4c50c9a5b76812 Mon Sep 17 00:00:00 2001 From: John Bodley Date: Tue, 21 Nov 2023 12:18:34 -0800 Subject: [PATCH] fix: Optimize fetching samples logic --- superset/views/datasource/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/superset/views/datasource/utils.py b/superset/views/datasource/utils.py index 9baabdcc54163..afed5f7fd2596 100644 --- a/superset/views/datasource/utils.py +++ b/superset/views/datasource/utils.py @@ -114,7 +114,7 @@ def get_samples( # pylint: disable=too-many-arguments,too-many-locals sample_data = samples_instance.get_payload()["queries"][0] if sample_data.get("status") == QueryStatus.FAILED: - QueryCacheManager.delete(sample_data.get("cache_key"), CacheRegion.DATA) + QueryCacheManager.delete(count_star_data.get("cache_key"), CacheRegion.DATA) raise DatasetSamplesFailedError(sample_data.get("error")) sample_data["page"] = page