From 5ab3acd587720b9b8e6a8cb45dd1a6ec73980ea5 Mon Sep 17 00:00:00 2001
From: "pre-commit-ci[bot]"
 <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Date: Tue, 5 Dec 2023 12:56:30 +0000
Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
---
 pvnet/models/multimodal/site_encoders/encoders.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/pvnet/models/multimodal/site_encoders/encoders.py b/pvnet/models/multimodal/site_encoders/encoders.py
index 98102b58..8a0b2f3f 100644
--- 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=}")