From ba9bc43ca60b6edf52a07c6b69bcc07e14f2a518 Mon Sep 17 00:00:00 2001
From: bxdd <bxddream@gmail.com>
Date: Thu, 16 Nov 2023 22:03:48 +0800
Subject: [PATCH] [ENH] make setup install full package

---
 setup.py | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/setup.py b/setup.py
index 705000d..8c6b36d 100644
--- a/setup.py
+++ b/setup.py
@@ -69,6 +69,7 @@ REQUIRED = [
     "transformers>=4.34.1",
     "portalocker>=2.0.0",
     "qpsolvers[clarabel]>=4.0.1",
+    "geatpy>=2.7.0;python_version<='3.11'",
 ]
 
 here = os.path.abspath(os.path.dirname(__file__))
@@ -105,6 +106,14 @@ if __name__ == "__main__":
                 "black==23.1.0",
                 "pre-commit",
             ],
+            "full": [
+                "torch>=1.11.0",
+                "torchvision>=0.15.1",
+                "torch-optimizer>=0.3.0",
+                "lightgbm>=3.3.0",
+                "torchvision>=0.15.1",
+                "sentence_transformers>=2.2.2",
+            ],
         },
         classifiers=[
             "Intended Audience :: Science/Research",