enable report title

Author: ahosler

ads/opctl/operator/lowcode/forecast/model/base_model.py

@@ -29,7 +29,6 @@
     seconds_to_datetime,
     write_data,
 )
-from ads.opctl.operator.lowcode.forecast.model.forecast_datasets import TestData
 from ads.opctl.operator.lowcode.forecast.utils import (
     _build_metrics_df,
     _build_metrics_per_horizon,
@@ -132,11 +131,10 @@ def generate_report(self):
 
                 if self.datasets.test_data is not None:
                     try:
-                        (
-                            self.test_eval_metrics,
-                            summary_metrics
-                        ) = self._test_evaluate_metrics(
-                            elapsed_time=elapsed_time,
+                        (self.test_eval_metrics, summary_metrics) = (
+                            self._test_evaluate_metrics(
+                                elapsed_time=elapsed_time,
+                            )
                         )
                         if not self.target_cat_col:
                             self.test_eval_metrics.rename(
@@ -155,9 +153,9 @@ def generate_report(self):
                     model_description,
                     other_sections,
                 ) = self._generate_report()
-
+                report_title = self.config.spec.report_title or "Forecast Report"
                 header_section = rc.Block(
-                    rc.Heading("Forecast Report", level=1),
+                    rc.Heading(report_title, level=1),
                     rc.Text(
                         f"You selected the {self.spec.model} model.\nBased on your dataset, you could have also selected any of the models: {SupportedModels.keys()}."
                     ),
@@ -471,7 +469,7 @@ def _save_report(
         result_df: pd.DataFrame,
         metrics_df: pd.DataFrame,
         test_metrics_df: pd.DataFrame,
-        test_data: pd.DataFrame,
+        # test_data: pd.DataFrame,
     ):
         """Saves resulting reports to the given folder."""
 

tests/operators/forecast/test_errors.py

@@ -835,9 +835,9 @@ def test_what_if_analysis(operator_setup, model):
     historical_data = pd.read_csv(historical_data_path, parse_dates=["Date"])
     historical_filtered = historical_data[historical_data["Date"] > "2013-03-01"]
     additional_data = pd.read_csv(additional_data_path, parse_dates=["Date"])
-    add_filtered = additional_data[additional_data['Date'] > "2013-03-01"]
-    add_filtered.to_csv(f'{additional_test_path}', index=False)
-    historical_filtered.to_csv(f'{historical_test_path}', index=False)
+    add_filtered = additional_data[additional_data["Date"] > "2013-03-01"]
+    add_filtered.to_csv(f"{additional_test_path}", index=False)
+    historical_filtered.to_csv(f"{historical_test_path}", index=False)
 
     yaml_i, output_data_path = populate_yaml(
         tmpdirname=tmpdirname,
@@ -893,5 +893,33 @@ def test_auto_select(operator_setup):
     report_path = f"{output_data_path}/report.html"
     assert os.path.exists(report_path), f"Report file not found at {report_path}"
 
+
+@pytest.mark.parametrize("model", ["prophet"])
+def test_report_title(operator_setup, model):
+    from ads.opctl.operator.lowcode.forecast.__main__ import operate
+    from ads.opctl.operator.lowcode.forecast.operator_config import (
+        ForecastOperatorConfig,
+    )
+
+    yaml_i = TEMPLATE_YAML.copy()
+    yaml_i["spec"]["horizon"] = 10
+    yaml_i["spec"]["model"] = model
+    yaml_i["spec"]["historical_data"] = {"format": "pandas"}
+    yaml_i["spec"]["target_column"] = TARGET_COL.name
+    yaml_i["spec"]["datetime_column"]["name"] = HISTORICAL_DATETIME_COL.name
+    yaml_i["spec"]["report_title"] = "Skibidi ADS Skibidi"
+    yaml_i["spec"]["output_directory"]["url"] = operator_setup
+
+    df = pd.concat([HISTORICAL_DATETIME_COL[:15], TARGET_COL[:15]], axis=1)
+    yaml_i["spec"]["historical_data"]["data"] = df
+    operator_config = ForecastOperatorConfig.from_dict(yaml_i)
+    results = operate(operator_config)
+    with open(os.path.join(operator_setup, "report.html")) as f:
+        for line in f:
+            if "Skibidi ADS Skibidi" in line:
+                return True
+        assert False, "Report Title was not set"
+
+
 if __name__ == "__main__":
     pass