From e227b655564f96a35f12f2811489908d143364a1 Mon Sep 17 00:00:00 2001
From: peterdudfield <peter.dudfield@hotmail.com>
Date: Thu, 5 Sep 2024 17:03:04 +0100
Subject: [PATCH] at print statment

---
 pvnet/models/base_model.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/pvnet/models/base_model.py b/pvnet/models/base_model.py
index dc86a879..c2354a2b 100644
--- a/pvnet/models/base_model.py
+++ b/pvnet/models/base_model.py
@@ -623,6 +623,8 @@ def _log_validation_results(self, batch, y_hat, accum_batch_num):
             y_i = y[i].detach().cpu().numpy()
             y_hat_i = y_hat[i].detach().cpu().numpy()
 
+            print(BatchKey._member_map_)
+
             time_utc_key = BatchKey[f"{self._target_key}_time_utc"]
             time_utc = batch[time_utc_key][i, -self.forecast_len :].detach().cpu().numpy()