fix: Optimize fetching samples logic (#26060)

This commit is contained in:
John Bodley
2023-11-21 13:26:02 -08:00
committed by GitHub
parent 97121465dd
commit bd8951e958

View File

@@ -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