Browse Source

[ENH] make setup install full package

tags/v0.3.2
bxdd 2 years ago
parent
commit
ba9bc43ca6
1 changed files with 9 additions and 0 deletions
  1. +9
    -0
      setup.py

+ 9
- 0
setup.py View File

@@ -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",


Loading…
Cancel
Save