From 7521c55b52e897b49839bc0967e5f084907b9bfd Mon Sep 17 00:00:00 2001
From: Wilson Beebe <wbeebe@uw.edu>
Date: Thu, 21 Dec 2023 10:36:00 -0800
Subject: [PATCH] lint fix

---
 src/tape/ensemble.py | 17 +++++++++++++----
 1 file changed, 13 insertions(+), 4 deletions(-)

diff --git a/src/tape/ensemble.py b/src/tape/ensemble.py
index 056e8813..d5e9ec21 100644
--- 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,
             )