Fix error on target key

openclimatefix · Jan 24, 2024 · 0d8d349 · 0d8d349
1 parent 468b0b3
commit 0d8d349
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pvnet/models/base_model.py b/pvnet/models/base_model.py
@@ -256,7 +256,7 @@ def __init__(
 
         self._optimizer = optimizer
         self._target_key_name = target_key
-        self._target_key = BatchKey[f"{self._target_key}"]
+        self._target_key = BatchKey[f"{target_key}"]
 
         # Model must have lr to allow tuning
         # This setting is only used when lr is tuned with callback