Skip to content

Commit 15e282a

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent 08c776e commit 15e282a

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

pvnet/models/multimodal/multimodal.py

+3-3
Original file line numberDiff line numberDiff line change
@@ -306,7 +306,7 @@ def forward(self, x):
306306
# Target is PV, so only take the history
307307
# Copy batch
308308
x_tmp = x.copy()
309-
x_tmp[BatchKey.pv] = x_tmp[BatchKey.pv][:, : self.history_len+1]
309+
x_tmp[BatchKey.pv] = x_tmp[BatchKey.pv][:, : self.history_len + 1]
310310
modes["pv"] = self.pv_encoder(x_tmp)
311311

312312
# *********************** GSP Data ************************************
@@ -329,7 +329,7 @@ def forward(self, x):
329329
else:
330330
# Have to be its own Batch format
331331
x_tmp = x.copy()
332-
x_tmp[BatchKey.wind] = x_tmp[BatchKey.wind][:, : self.history_len+1]
332+
x_tmp[BatchKey.wind] = x_tmp[BatchKey.wind][:, : self.history_len + 1]
333333
# This needs to be a Batch as input
334334
modes["wind"] = self.wind_encoder(x_tmp)
335335

@@ -339,7 +339,7 @@ def forward(self, x):
339339
modes["sensor"] = self.sensor_encoder(x)
340340
else:
341341
x_tmp = x.copy()
342-
x_tmp[BatchKey.sensor] = x_tmp[BatchKey.sensor][:, : self.history_len+1]
342+
x_tmp[BatchKey.sensor] = x_tmp[BatchKey.sensor][:, : self.history_len + 1]
343343
# This needs to be a Batch as input
344344
modes["sensor"] = self.sensor_encoder(x_tmp)
345345

0 commit comments

Comments
 (0)