diff --git a/pvnet/models/base_model.py b/pvnet/models/base_model.py
index 5dfb553d..cc7aecae 100644
--- a/pvnet/models/base_model.py
+++ b/pvnet/models/base_model.py
@@ -637,7 +637,7 @@ def _log_validation_results(self, batch, y_hat, accum_batch_num):
             y_i = y[i]
             if self.use_quantile_regression:
                 idx = self.output_quantiles.index(0.5)
-                y_hat_i = y_hat[i,idx]
+                y_hat_i = y_hat[i, idx]
             else:
                 y_hat_i = y_hat[i]
             time_utc_i = time_utc[i]