Dimensions overview fix

teamdigitale · Jul 19, 2023 · 6f14e42 · 6f14e42
1 parent 5859a07
commit 6f14e42
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/pages/0_👀_Lookup.py b/pages/0_👀_Lookup.py
@@ -45,10 +45,11 @@ def load_dimensions(metabase2datasets: pd.DataFrame) -> pd.Series:
         key="_selected_codes",
         multiedit=True,
     )
-    if selected_codes is not None:
-        selected_codes_mask = selected_codes["selected"].values
-    else:
-        selected_codes_mask = codes["selected"].values
+    selected_codes_mask = (
+        codes["selected"].values
+        if "_selected_codes_data" not in session
+        else session["_selected_codes_data"]["selected"].values
+    )
 
     # TODO  Retrieve selected dimension after page change
     st.markdown("Selected dimension overview:")