diff --git a/pvnet/models/base_model.py b/pvnet/models/base_model.py index 14e4d48d..452c79e3 100644 --- a/pvnet/models/base_model.py +++ b/pvnet/models/base_model.py @@ -370,7 +370,7 @@ def _calculate_val_losses(self, y, y_hat): # Take median value for remaining metric calculations y_hat = self._quantiles_to_prediction(y_hat) - common_metrics_each_step = common_metrics(predictions=y_hat, targets=y) + common_metrics_each_step = common_metrics(predictions=y_hat, target=y) mse_each_step = common_metrics_each_step["rmse"] ** 2 mae_each_step = common_metrics_each_step["mae"]