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

for more information, see https://pre-commit.ci
openclimatefix · Feb 8, 2024 · 74aae2b · 74aae2b
1 parent 0a10da2
commit 74aae2b
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/pvnet/models/multimodal/multimodal.py b/pvnet/models/multimodal/multimodal.py
@@ -343,7 +343,11 @@ def forward(self, x):
 
         if self.include_sun:
             sun = torch.cat(
-                (x[BatchKey[f"{self.target_key_name}_solar_azimuth"]], x[BatchKey[f"{self.target_key_name}_solar_elevation"]]), dim=1
+                (
+                    x[BatchKey[f"{self.target_key_name}_solar_azimuth"]],
+                    x[BatchKey[f"{self.target_key_name}_solar_elevation"]],
+                ),
+                dim=1,
             ).float()
             sun = self.sun_fc1(sun)
             modes["sun"] = sun