diff --git a/pvnet_app/forecast_compiler.py b/pvnet_app/forecast_compiler.py index 55bbdbe..bdcfafd 100644 --- a/pvnet_app/forecast_compiler.py +++ b/pvnet_app/forecast_compiler.py @@ -161,6 +161,7 @@ def predict_batch(self, batch: NumpyBatch) -> None: # We only need elevation mask for forecasted values, not history elevation = elevation[:, -preds.shape[1] :] sun_down_mask = elevation < MIN_DAY_ELEVATION + self.log_info.info(f'{sun_down_mask=}') # Store predictions internally self.normed_preds += [preds]