black

alan-turing-institute · Sep 19, 2023 · e28ba4b · e28ba4b
1 parent 9207599
commit e28ba4b
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/deepsensor/data/task.py b/deepsensor/data/task.py
@@ -148,7 +148,9 @@ def remove_nans_from_task_Y_t_if_present(self):
             return self
 
         # NaNs present in self - remove NaNs
-        for i, (X, Y, Y_t_nans) in enumerate(zip(self["X_t"], self["Y_t"], Y_t_nans_list)):
+        for i, (X, Y, Y_t_nans) in enumerate(
+            zip(self["X_t"], self["Y_t"], Y_t_nans_list)
+        ):
             if B.any(Y_t_nans):
                 if isinstance(X, tuple):
                     # Gridded data