Added extras for synapse (microsoft#916)

* Added extras for synapse * Update Installation doc
whiskyboy · Feb 17, 2023 · 8c6d608 · 8c6d608
1 parent a8a89c2
commit 8c6d608
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/setup.py b/setup.py
@@ -114,6 +114,7 @@
         ],
         "benchmark": ["catboost>=0.26", "psutil==5.8.0", "xgboost==1.3.3"],
         "openai": ["openai==0.23.1", "diskcache", "optuna==2.8.0"],
+        "synapse": ["joblibspark>=0.5.0", "optuna==2.8.0", "pyspark>=3.0.0"],
     },
     classifiers=[
         "Programming Language :: Python :: 3",

diff --git a/website/docs/Installation.md b/website/docs/Installation.md
@@ -81,6 +81,12 @@ pip install flaml[nni]
 pip install flaml[blendsearch]
 ```
 
+* synapse
+> *To install flaml in Azure Synapse and similar cloud platform*
+```bash
+pip install flaml[synapse]
+```
+
 #### Test and Benchmark
 
 * test