add install dbconnect

mosaicml · Dec 5, 2023 · a0c4249 · a0c4249
1 parent 89197fc
commit a0c4249
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/setup.py b/setup.py
@@ -86,6 +86,7 @@
 extra_deps['databricks'] = [
     'mosaicml[databricks]>=0.17.1,<0.18',
     'databricks-sql-connector>=3, <4',
+    'databricks-connect==14.0.0'
 ]
 
 extra_deps['tensorboard'] = [