diff --git a/src/schedm_mad/remotes/one_drs/lib/optimizer_parser.py b/src/schedm_mad/remotes/one_drs/lib/optimizer_parser.py index cbd7c33a6a..aa07f09a4b 100644 --- a/src/schedm_mad/remotes/one_drs/lib/optimizer_parser.py +++ b/src/schedm_mad/remotes/one_drs/lib/optimizer_parser.py @@ -972,7 +972,7 @@ def get_system_ds(self, host_id): def _apply_predictive_adjustment( self, current: float, forecast: float = None ) -> float: - predictive = self.config.get("predictive", 0) + predictive = self.config.get("PREDICTIVE") or 0 if predictive > 0 and forecast > 0: return current * (1 - predictive) + forecast * predictive return current