Add logic to skip hive lock wait if retries are set to 0

Author: MoIMC

pyiceberg/catalog/hive.py

@@ -488,6 +488,9 @@ def _do_wait_for_lock() -> LockResponse:
                     f"Failed to check lock for {database_name}.{table_name}, lock state: {response.state}"
                 )
 
+        if not self._lock_check_retries:
+            raise CommitFailedException(f"Lock was not acquired for commit and retries are set to `0`.")
+
         return _do_wait_for_lock()
 
     def commit_table(

pyiceberg/table/__init__.py

@@ -825,7 +825,7 @@ def _error_callback(state: RetryCallState):
         @wraps(self.commit_transaction)
         @retry(
             wait=wait_random_exponential(min=min_wait_ms / 1000, max=max_wait_ms / 1000),
-            stop=stop_after_attempt(num_retries),
+            stop=stop_after_attempt(num_retries + 1),
             retry=retry_if_exception_type(CommitFailedException),
             before=_before_attempt,
             retry_error_callback=_error_callback,