only change the private method

Author: HarrisonMarcRose

stitchclient/client.py

@@ -112,7 +112,7 @@ def moving_average_bytes_per_record(self):
             num_records += stats.num_records
             num_bytes += stats.num_bytes
 
-        return num_bytes // num_records if num_records else None
+        return num_bytes // num_records
 
     def push(self, message, callback_arg=None):
         """message should be a dict recognized by the Stitch Import API.
@@ -151,16 +151,20 @@ def _take_batch(self, min_records):
         self._buffer.clear()
         return result
 
+
     def _send_batch(self, batch):
         for body, callback_args in partition_batch(batch, self.max_batch_size_bytes):
             self._send(body, callback_args)
 
-        if self.moving_average_bytes_per_record():
+        try:
+            moving_average = self.moving_average_bytes_per_record()
             self.target_messages_per_batch = \
                 min(self.max_messages_per_batch,
-                    0.8 * (self.max_batch_size_bytes / self.moving_average_bytes_per_record()))
-        else:
-            self.target_messages_per_batch = self.max_messages_per_batch
+                    0.8 * (self.max_batch_size_bytes / moving_average))
+        except ZeroDivisionError:
+            # Handle the case where there are no records
+            pass
+
 
     def _stitch_request(self, body):
         headers = {'Authorization': 'Bearer {}'.format(self.token),