Merge branch 'aarthy/qa-fixes' into 'enterprise'

More QA fixes

See merge request dkinternal/testgen/dataops-testgen!332

Author: ci bot

testgen/common/models/profiling_run.py

@@ -160,14 +160,13 @@ def select_summary(
                 SUM(
                     CASE
                         WHEN COALESCE(profile_anomaly_results.disposition, 'Confirmed') = 'Confirmed'
-                        AND profile_anomaly_types.issue_likelihood = 'Possible' THEN 1
+                        AND profile_anomaly_types.issue_likelihood IN ('Possible', 'Potential PII') THEN 1
                         ELSE 0
                     END
                 ) AS possible_ct,
                 SUM(
                     CASE
-                        WHEN COALESCE(profile_anomaly_results.disposition, 'Confirmed') IN ('Dismissed', 'Inactive')
-                        AND profile_anomaly_types.issue_likelihood <> 'Potential PII' THEN 1
+                        WHEN COALESCE(profile_anomaly_results.disposition, 'Confirmed') IN ('Dismissed', 'Inactive') THEN 1
                         ELSE 0
                     END
                 ) AS dismissed_ct

testgen/common/models/table_group.py

@@ -143,14 +143,13 @@ def select_summary(cls, project_code: str, for_dashboard: bool = False) -> Itera
                 SUM(
                     CASE
                         WHEN COALESCE(latest_anomalies.disposition, 'Confirmed') = 'Confirmed'
-                        AND anomaly_types.issue_likelihood = 'Possible' THEN 1
+                        AND anomaly_types.issue_likelihood IN ('Possible', 'Potential PII') THEN 1
                         ELSE 0
                     END
                 ) AS possible_ct,
                 SUM(
                     CASE
-                        WHEN COALESCE(latest_anomalies.disposition, 'Confirmed') IN ('Dismissed', 'Inactive')
-                        AND anomaly_types.issue_likelihood <> 'Potential PII' THEN 1
+                        WHEN COALESCE(latest_anomalies.disposition, 'Confirmed') IN ('Dismissed', 'Inactive') THEN 1
                         ELSE 0
                     END
                 ) AS dismissed_ct

testgen/ui/navigation/router.py

@@ -30,6 +30,11 @@ def run(self) -> None:
 
         current_page = st.navigation(streamlit_pages, position="hidden")
 
+        if not session.initialized:
+            # Clear cache on initial load or page refresh
+            st.cache_data.clear()
+            session.initialized = True
+
         # This hack is needed because the auth cookie is not set if navigation happens immediately after login
         # We have to navigate on the next run
         if session.auth.logging_in:

testgen/ui/session.py

@@ -23,6 +23,7 @@ class TestgenSession(Singleton):
     # streamlit_authenticator sets this attribute implicitly
     authentication_status: bool
 
+    initialized: bool
     page_pending_cookies: st.Page  # type: ignore
     page_pending_login: str
     page_args_pending_login: dict