Skip to content

Commit 74aae2b

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent 0a10da2 commit 74aae2b

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

pvnet/models/multimodal/multimodal.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -343,7 +343,11 @@ def forward(self, x):
343343

344344
if self.include_sun:
345345
sun = torch.cat(
346-
(x[BatchKey[f"{self.target_key_name}_solar_azimuth"]], x[BatchKey[f"{self.target_key_name}_solar_elevation"]]), dim=1
346+
(
347+
x[BatchKey[f"{self.target_key_name}_solar_azimuth"]],
348+
x[BatchKey[f"{self.target_key_name}_solar_elevation"]],
349+
),
350+
dim=1,
347351
).float()
348352
sun = self.sun_fc1(sun)
349353
modes["sun"] = sun

0 commit comments

Comments
 (0)