use list comprehension

Author: sojingle

src/amplitude/processor.py

@@ -22,8 +22,9 @@ def process_response(self, res, events):
                 self.callback(events, res.code, res.error)
                 self.log(events, res.code, res.error)
             else:
-                # Only reduce if batch was at or below current limit
-                # This prevents multiple reductions from same flush operation
+                # Reduce only if batch didn't exceed current limit (was expected to work).
+                # Batches larger than limit are from old limits already deemed too large,
+                # so failing again doesn't provide new information - skip to avoid over-reduction.
                 if len(events) <= self.configuration.flush_queue_size:
                     self.configuration._increase_flush_divider()
                 self.push_to_storage(events, 0, res)

src/amplitude/worker.py

@@ -46,13 +46,8 @@ def flush(self):
             return None
 
         batch_size = self.configuration.flush_queue_size
-        batches = []
-        for i in range(0, len(events), batch_size):
-            batches.append(events[i:i + batch_size])
-
-        batch_futures = []
-        for batch in batches:
-            batch_futures.append(self.threads_pool.submit(self.send, batch))
+        batches = [events[i:i + batch_size] for i in range(0, len(events), batch_size)]
+        batch_futures = [self.threads_pool.submit(self.send, batch) for batch in batches]
 
         if len(batch_futures) == 1:
             return batch_futures[0]
@@ -76,9 +71,6 @@ def wait_for_all():
             if errors:
                 raise errors[0]
 
-            self.configuration.logger.info(
-                f"Flush completed: {len(batch_futures)} batches sent"
-            )
             return None
 
         return self.threads_pool.submit(wait_for_all)