lint fix

lincc-frameworks · Dec 21, 2023 · 7521c55 · 7521c55
1 parent 9f94c69
commit 7521c55
Showing 1 changed file with 13 additions and 4 deletions.
diff --git a/src/tape/ensemble.py b/src/tape/ensemble.py
@@ -1008,7 +1008,18 @@ def bin_sources(
         self.source.set_dirty(True)
         return self
 
-    def batch(self, func, *args, meta=None, by_band=False, use_map=True, on=None, sort_by_time=True, label="", **kwargs):
+    def batch(
+        self,
+        func,
+        *args,
+        meta=None,
+        by_band=False,
+        use_map=True,
+        on=None,
+        sort_by_time=True,
+        label="",
+        **kwargs,
+    ):
         """Run a function from tape.TimeSeries on the available ids
 
         Parameters
@@ -1175,9 +1186,7 @@ def convert_and_sort(y, sort, id_col, time_col):
             )
         else:  # use groupby
             batch = source_to_batch.groupby(on, group_keys=False).apply(
-                lambda x: func(
-                    *convert_and_sort(x, sort_by_time, id_col, time_col), **kwargs
-                ),
+                lambda x: func(*convert_and_sort(x, sort_by_time, id_col, time_col), **kwargs),
                 meta=meta,
             )