Merge branch 'main' into fix/es908737

databrickslabs · Oct 27, 2023 · a7e9e0b · a7e9e0b
2 parents cc8f19d + 226f078
commit a7e9e0b
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/src/databricks/labs/ucx/framework/dashboards.py b/src/databricks/labs/ucx/framework/dashboards.py
@@ -175,10 +175,9 @@ def _installed_query_state(self):
             self._state = json.load(self._ws.workspace.download(self._query_state))
             to_remove = []
             for k, v in self._state.items():
-                _, name = k.split(":")
-                if k == "dashboard_id":
+                if k.endswith("dashboard_id"):
                     continue
-                if name != "query_id":
+                if not k.endswith("query_id"):
                     continue
                 try:
                     self._ws.queries.get(v)
@@ -215,7 +214,7 @@ def _store_query_state(self, queries: dict[str, list[SimpleQuery]]):
             if k in desired_keys:
                 new_state[k] = v
                 continue
-            _, name = k.split(":")
+            name = k if ":" not in k else k.split(":")[-1]
             if name not in destructors:
                 continue
             try: