Add TODO

databricks · Nov 14, 2024 · 6d2c183 · 6d2c183
1 parent 21c797b
commit 6d2c183
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/databricks/sdk/_base_client.py b/databricks/sdk/_base_client.py
@@ -270,6 +270,9 @@ def _perform(self,
         if error is not None:
             # If the request body is a seekable stream, rewind it so that it is ready
             # to be read again in case of a retry.
+            #
+            # TODO: This should be moved into a "before-retry" hook to avoid one
+            # unnecessary seek on the last failed retry before aborting.
             if self._is_seekable_stream(data):
                 data.seek(initial_data_position)
             raise error from None