fix: Optimize fetching samples logic (apache#26060)

Ortege-xyz · Jan 22, 2024 · 99e3e40 · 99e3e40
1 parent 83ee9c4
commit 99e3e40
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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