Skip to content

Commit

Permalink
Created issue for TODO
Browse files Browse the repository at this point in the history
  • Loading branch information
JCZuurmond committed Dec 10, 2024
1 parent bd0ef22 commit ed9612e
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions src/databricks/labs/ucx/assessment/dashboards.py
Original file line number Diff line number Diff line change
Expand Up @@ -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)

Expand All @@ -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
Expand Down

0 comments on commit ed9612e

Please sign in to comment.