From 5abfce3b2ab02f9f861502c3b904213d8c1dbead Mon Sep 17 00:00:00 2001 From: Cor Zuurmond Date: Tue, 10 Dec 2024 10:15:33 +0100 Subject: [PATCH] Created issue for TODO https://github.com/databrickslabs/ucx/issues/3415 --- src/databricks/labs/ucx/assessment/dashboards.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/databricks/labs/ucx/assessment/dashboards.py b/src/databricks/labs/ucx/assessment/dashboards.py index f7fbaf103b..700a93bfbf 100644 --- a/src/databricks/labs/ucx/assessment/dashboards.py +++ b/src/databricks/labs/ucx/assessment/dashboards.py @@ -244,7 +244,7 @@ def _list_all_legacy_queries(self) -> Iterator[LegacyQuery]: yield from self._get_legacy_queries(*self._include_query_ids) else: try: - yield from self._ws.queries_legacy.list() # TODO: Update this to non-legacy query + yield from self._ws.queries_legacy.list() except DatabricksError as e: logger.warning("Cannot list Redash queries", exc_info=e) @@ -266,7 +266,7 @@ def _get_legacy_queries(self, *query_ids: str) -> Iterator[LegacyQuery]: def _get_legacy_query(self, query_id: str) -> LegacyQuery | None: """Get a legacy query.""" try: - return self._ws.queries_legacy.get(query_id) # TODO: Update this to non-legacy query + return self._ws.queries_legacy.get(query_id) except DatabricksError as e: logger.warning(f"Cannot get Redash query: {query_id}", exc_info=e) return None