Route AnalysisNotApplicableStateError to NotApplicableAnalysisCard (#5051)

Summary:

Update error_card_from_analysis_e() in analysis.py to route AnalysisNotApplicableStateError exceptions to NotApplicableStateAnalysisCard instead of ErrorAnalysisCard.

Differential Revision: D96248248

Author: shrutipatel31

ax/analysis/analysis.py

@@ -19,6 +19,7 @@
     AnalysisCardBase,
     AnalysisCardGroup,
     ErrorAnalysisCard,
+    NotApplicableStateAnalysisCard,
 )
 from ax.core.experiment import Experiment
 from ax.exceptions.analysis import AnalysisNotApplicableStateError
@@ -29,6 +30,12 @@
 
 logger: Logger = get_logger(__name__)
 
+NOT_APPLICABLE_STATE_SUBTITLE: str = (
+    "This analysis is temporarily unavailable. It will become available "
+    "as your experiment progresses (e.g., after collecting more data, "
+    "running more trials, or fitting a model)."
+)
+
 
 class Analysis(Protocol):
     """
@@ -202,7 +209,7 @@ def display_cards(
 
 def error_card_from_analysis_e(
     analysis_e: AnalysisE,
-) -> ErrorAnalysisCard:
+) -> ErrorAnalysisCard | NotApplicableStateAnalysisCard:
     analysis_name = analysis_e.analysis.__class__.__name__
     exception_name = analysis_e.exception.__class__.__name__
 
@@ -213,6 +220,18 @@ def error_card_from_analysis_e(
         if (exception_message := str(analysis_e.exception))
         else f"{exception_name} encountered while computing {analysis_name}."
     )
+    if isinstance(analysis_e.exception, AnalysisNotApplicableStateError):
+        # AnalysisNotApplicableStateError gets rendered as a
+        # NotApplicableStateAnalysisCard
+        return NotApplicableStateAnalysisCard(
+            name=analysis_name,
+            title=f"{analysis_name} -- Not Available Yet",
+            subtitle=NOT_APPLICABLE_STATE_SUBTITLE,
+            df=pd.DataFrame(),
+            blob=exception_message
+            if exception_message
+            else f"{exception_name} encountered.",
+        )
 
     return ErrorAnalysisCard(
         name=analysis_name,

ax/analysis/tests/test_analysis.py

@@ -1,3 +1,4 @@
+#!/usr/bin/env python3
 # Copyright (c) Meta Platforms, Inc. and affiliates.
 #
 # This source code is licensed under the MIT license found in the
@@ -7,6 +8,8 @@
 
 from ax.analysis.analysis import AnalysisE, error_card_from_analysis_e
 from ax.analysis.plotly.parallel_coordinates import ParallelCoordinatesPlot
+from ax.core.analysis_card import NotApplicableStateAnalysisCard
+from ax.exceptions.analysis import AnalysisNotApplicableStateError
 from ax.utils.common.testutils import TestCase
 
 
@@ -35,3 +38,17 @@ def test_error_card_from_analysis_e(self) -> None:
                 self.assertEqual(card.title, "ParallelCoordinatesPlot Error")
                 self.assertEqual(card.subtitle, expected_subtitle)
                 self.assertIn("ValueError", card.blob)
+
+    def test_not_applicable_error_produces_not_applicable_state_card(self) -> None:
+        analysis_e = AnalysisE(
+            message="Analysis not applicable",
+            exception=AnalysisNotApplicableStateError("Experiment has no data."),
+            analysis=ParallelCoordinatesPlot(),
+        )
+
+        card = error_card_from_analysis_e(analysis_e)
+
+        self.assertIsInstance(card, NotApplicableStateAnalysisCard)
+        self.assertEqual(card.name, "ParallelCoordinatesPlot")
+        self.assertEqual(card.title, "ParallelCoordinatesPlot -- Not Available Yet")
+        self.assertEqual(card.blob, "Experiment has no data.")

ax/api/tests/test_client.py

@@ -1278,10 +1278,12 @@ def test_compute_analyses(self) -> None:
 
         self.assertEqual(len(cards), 1)
         self.assertEqual(cards[0].name, "ParallelCoordinatesPlot")
-        self.assertEqual(cards[0].title, "ParallelCoordinatesPlot Error")
+        self.assertEqual(cards[0].title, "ParallelCoordinatesPlot -- Not Available Yet")
         self.assertEqual(
             cards[0].subtitle,
-            "AnalysisNotApplicableStateError: Experiment has no trials.",
+            "This analysis is temporarily unavailable. It will become available as "
+            "your experiment progresses (e.g., after collecting more data, running "
+            "more trials, or fitting a model).",
         )
 
         for trial_index, _ in client.get_next_trials(max_trials=1).items():