From f36b5509c38d072782648ccc8959866c0e9b0a6f Mon Sep 17 00:00:00 2001 From: Jacob Bieker Date: Tue, 5 Dec 2023 10:29:54 +0000 Subject: [PATCH] Change default split key for testing WindNet --- pvnet/data/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pvnet/data/utils.py b/pvnet/data/utils.py index be173f60..847c349a 100644 --- 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]