Add missing dependencies due to autogen's installation issue.

microsoft · Sep 20, 2024 · e2d90bb · e2d90bb
1 parent 8f9de52
commit e2d90bb
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -10,7 +10,9 @@
 
 install_requires = [
     "pyautogen>=0.2.31",
-    "graphviz>=0.20.1"
+    "graphviz>=0.20.1",
+    "scikit-learn",
+    "xgboost",
 ]
 
 setuptools.setup(