Change default split key for testing WindNet

openclimatefix · Dec 5, 2023 · f36b550 · f36b550
1 parent f44d89c
commit f36b550
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pvnet/data/utils.py b/pvnet/data/utils.py
@@ -24,7 +24,7 @@ def batch_to_tensor(batch):
     return batch
 
 
-def split_batches(batch, splitting_key=BatchKey.gsp):
+def split_batches(batch, splitting_key=BatchKey.sensor):
     """Splits a single batch of data."""
 
     n_samples = batch[splitting_key].shape[0]