Include INFO status healthcheck cards in overview to save to db

Summary: In order for the INFO healthcheck cards to be shown in the Ax UI, they need to be saved to the db. This diff adds those cards to overview cards so they can be saved appropriately.

Differential Revision: D96492464

Author: shrutipatel31

ax/analysis/overview.py

@@ -20,7 +20,10 @@
     ConstraintsFeasibilityAnalysis,
 )
 from ax.analysis.healthcheck.early_stopping_healthcheck import EarlyStoppingAnalysis
-from ax.analysis.healthcheck.healthcheck_analysis import HealthcheckAnalysisCard
+from ax.analysis.healthcheck.healthcheck_analysis import (
+    HealthcheckAnalysisCard,
+    HealthcheckStatus,
+)
 from ax.analysis.healthcheck.metric_fetching_errors import MetricFetchingErrorsAnalysis
 from ax.analysis.healthcheck.predictable_metrics import PredictableMetricsAnalysis
 from ax.analysis.healthcheck.search_space_analysis import SearchSpaceAnalysis
@@ -268,7 +271,12 @@ def compute(
         non_passing_health_checks = [
             card
             for card in health_check_cards
-            if (isinstance(card, HealthcheckAnalysisCard) and not card.is_passing())
+            if (
+                isinstance(card, HealthcheckAnalysisCard)
+                and (
+                    not card.is_passing() or card.get_status() == HealthcheckStatus.INFO
+                )
+            )
             or isinstance(card, ErrorAnalysisCard)
         ]
 

ax/analysis/tests/test_overview.py

@@ -11,6 +11,10 @@
 import pandas as pd
 from ax.adapter.base import Adapter
 from ax.adapter.registry import Generators
+from ax.analysis.healthcheck.healthcheck_analysis import (
+    create_healthcheck_analysis_card,
+    HealthcheckStatus,
+)
 from ax.analysis.insights import InsightsAnalysis
 from ax.analysis.overview import OverviewAnalysis
 from ax.analysis.plotly.arm_effects import ArmEffectsPlot
@@ -447,3 +451,47 @@ def test_insights_analysis_single_parameter(self) -> None:
         # Check that none of the cards are error cards
         for card in all_cards:
             self.assertNotIsInstance(card, ErrorAnalysisCard)
+
+    def test_info_status_healthcheck_cards_in_overview(self) -> None:
+        """Test that healthcheck cards with INFO status are included in overview."""
+        # Create healthcheck cards with different statuses
+        pass_card = create_healthcheck_analysis_card(
+            name="PassCheck",
+            title="Pass Healthcheck",
+            subtitle="This check passed",
+            df=pd.DataFrame(),
+            status=HealthcheckStatus.PASS,
+        )
+        info_card = create_healthcheck_analysis_card(
+            name="InfoCheck",
+            title="Info Healthcheck",
+            subtitle="This is informational",
+            df=pd.DataFrame(),
+            status=HealthcheckStatus.INFO,
+        )
+        warning_card = create_healthcheck_analysis_card(
+            name="WarningCheck",
+            title="Warning Healthcheck",
+            subtitle="This is a warning",
+            df=pd.DataFrame(),
+            status=HealthcheckStatus.WARNING,
+        )
+        fail_card = create_healthcheck_analysis_card(
+            name="FailCheck",
+            title="Fail Healthcheck",
+            subtitle="This check failed",
+            df=pd.DataFrame(),
+            status=HealthcheckStatus.FAIL,
+        )
+
+        # INFO cards should be considered passing
+        self.assertTrue(info_card.is_passing())
+        self.assertTrue(pass_card.is_passing())
+        self.assertFalse(warning_card.is_passing())
+        self.assertFalse(fail_card.is_passing())
+
+        # Verify INFO status is correctly retrieved
+        self.assertEqual(info_card.get_status(), HealthcheckStatus.INFO)
+        self.assertEqual(pass_card.get_status(), HealthcheckStatus.PASS)
+        self.assertEqual(warning_card.get_status(), HealthcheckStatus.WARNING)
+        self.assertEqual(fail_card.get_status(), HealthcheckStatus.FAIL)