diff --git a/setup.py b/setup.py index 2f8d41de48..d4765d7abe 100644 --- a/setup.py +++ b/setup.py @@ -225,13 +225,13 @@ def package_files(prefix: str, directory: str, extension: str): extra_deps['mlflow'] = [ 'mlflow>=2.14.1,<3.0', - 'databricks-sdk==0.32.0', + 'databricks-sdk==0.33.0', 'pynvml>=11.5.0,<12', ] extra_deps['pandas'] = ['pandas>=2.0.0,<3.0'] -extra_deps['databricks'] = ['databricks-sdk==0.32.0'] +extra_deps['databricks'] = ['databricks-sdk==0.33.0'] extra_deps['all'] = {dep for deps in extra_deps.values() for dep in deps}