Add databricks dependency (#717)

mosaicml · Nov 6, 2023 · 2b74cb2 · 2b74cb2
1 parent 1d504c8
commit 2b74cb2
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/setup.py b/setup.py
@@ -83,6 +83,10 @@
     'hf_transfer==0.1.3',
 ]
 
+extra_deps['databricks'] = [
+    'mosaicml[databricks]',
+]
+
 extra_deps['tensorboard'] = [
     'mosaicml[tensorboard]>=0.16.1,<0.17',
 ]