Merge pull request #5200 from Speshman/feat/4907-expose-load-ml-source

Expose load_ml_source on the ML load stats entity

Author: springfall2008

apps/predbat/load_ml_component.py

@@ -1152,6 +1152,7 @@ def _publish_entity(self):
                 "model_age_hours": round(model_age_hours, 1) if model_age_hours is not None else None,
                 "training_days": self.load_data_age_days,
                 "status": self.model_status,
+                "load_ml_source": self.ml_source,
                 "model_version": MODEL_VERSION,
                 "epochs_trained": self.predictor.epochs_trained if self.predictor else 0,
                 "friendly_name": "ML Load Stats",

apps/predbat/tests/test_load_ml.py

@@ -2481,7 +2481,7 @@ def get_arg(self, key, default=None, indirect=True, combine=False, attribute=Non
     print("  Testing _publish_entity:")
     mock_base = MockBase()
 
-    component = LoadMLComponent(mock_base, load_ml_enable=True)
+    component = LoadMLComponent(mock_base, load_ml_enable=True, load_ml_source=False)
 
     # Mock dashboard_item to capture calls
     def mock_dashboard_item(entity_id, state, attributes, app):
@@ -2572,6 +2572,9 @@ def mock_dashboard_item(entity_id, state, attributes, app):
     assert "status" in attrs2, "status should be in attributes"
     assert attrs2["status"] == "active", f"Expected status 'active', got {attrs2['status']}"
 
+    # Check load_ml_source
+    assert attrs2["load_ml_source"] is False, f"Expected load_ml_source False, got {attrs2['load_ml_source']}"
+
     # Check model_version
     assert "model_version" in attrs2, "model_version should be in attributes"
     from load_predictor import MODEL_VERSION

docs/load-ml.md

@@ -455,6 +455,8 @@ ML Component: Validation MAE: 0.3245 kWh
 
 `sensor.predbat_load_ml_stats` publishes three error figures, and they measure different things:
 
+The `load_ml_source` attribute reports whether Load ML is selected as Predbat's load forecast source.
+
 | Attribute | What it measures |
 |-----------|------------------|
 | `mae_kwh` | Teacher-forced, one step ahead with real data in the lookback. Stays small even when the long-range forecast is useless, so do not read it as overall accuracy. |