[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
openclimatefix · Dec 5, 2023 · 5ab3acd · 5ab3acd
1 parent 3c7b7aa
commit 5ab3acd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pvnet/models/multimodal/site_encoders/encoders.py b/pvnet/models/multimodal/site_encoders/encoders.py
@@ -327,7 +327,7 @@ def __init__(
 
     def _encode_query(self, x):
         # Select the first one
-        gsp_ids = x[BatchKey.sensor_id][:,0,:].squeeze().int()
+        gsp_ids = x[BatchKey.sensor_id][:, 0, :].squeeze().int()
         print(f"{gsp_ids.shape=}")
         query = self.sensor_id_embedding(gsp_ids).unsqueeze(1)
         print(f"{query.shape=}")