📌 switch from using retry to tenacity (#710)

* switch to tenacity

* fix import order

Author: renardeinside

CHANGELOG.md

@@ -10,11 +10,17 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 [Please read through the Keep a Changelog (~5min)](https://keepachangelog.com/en/1.0.0/).
 ## [UNRELEASED] - YYYY-MM-DD
 
-# Fixed
+## [0.8.9] - 2022-03-06
+
+### Fixed
 
 - 🔨 fix `dbx deploy --no-package` when `--no-rebuild` is not specified
 - 🔗 broken links in the docs
 
+### Changed
+
+- 📌 switch from using `retry` to `tenacity`
+
 ## [0.8.8] - 2022-02-22
 
 # Fixed

dbx/__init__.py

@@ -1 +1 @@
-__version__ = "0.8.8"
+__version__ = "0.8.9"

dbx/api/client_provider.py

@@ -3,7 +3,7 @@
 
 import requests
 from databricks_cli.sdk import ApiClient
-from retry import retry
+from tenacity import retry, wait_exponential, stop_after_attempt
 
 from dbx.api.auth import AuthConfigProvider
 
@@ -33,7 +33,7 @@ def get_context_status(self, payload):
 
     # sometimes cluster is already in the status="RUNNING", however it couldn't yet provide execution context
     # to make the execute command stable is such situations, we add retry handler.
-    @retry(tries=10, delay=5, backoff=5)
+    @retry(wait=wait_exponential(multiplier=1, min=2, max=5), stop=stop_after_attempt(5))
     def create_context(self, payload):
         result = self.v1_client.perform_query(method="POST", path="/contexts/create", data=payload)
         return result

dbx/utils/file_uploader.py

@@ -4,7 +4,7 @@
 from typing import Optional, Tuple
 
 import mlflow
-from retry import retry
+from tenacity import retry, wait_exponential, stop_after_attempt
 
 from dbx.api.context import RichExecutionContextClient
 from dbx.utils import dbx_echo
@@ -67,7 +67,7 @@ class MlflowFileUploader(AbstractFileUploader):
     """
 
     @staticmethod
-    @retry(tries=3, delay=1, backoff=0.3)
+    @retry(wait=wait_exponential(multiplier=1, min=2, max=5), stop=stop_after_attempt(5))
     def _upload_file(file_path: Path):
         posix_path = PurePosixPath(file_path.as_posix())
         parent = str(posix_path.parent) if str(posix_path.parent) != "." else None

setup.py

@@ -9,10 +9,10 @@
 
 INSTALL_REQUIRES = [
     # to use Databricks and MLflow APIs
-    "retry>=0.9.2, <1.0.0",
     "requests>=2.24.0, <3.0.0",
     "mlflow-skinny>=1.28.0,<3.0.0",
     "databricks-cli>=0.17,<0.18",
+    "tenacity>=8.2.2,<=9.0.0",
     # CLI interface
     "click>=8.1.0,<9.0.0",
     "rich==12.6.0",