|
|
|
@@ -51,31 +51,31 @@ def get_platform(): |
|
|
|
# What packages are required for this module to be executed? |
|
|
|
# `estimator` may depend on other packages. In order to reduce dependencies, it is not written here. |
|
|
|
REQUIRED = [ |
|
|
|
"numpy>=1.20.0", |
|
|
|
"pandas>=0.25.1", |
|
|
|
"scipy>=1.0.0", |
|
|
|
"matplotlib>=3.1.3", |
|
|
|
"torch>=1.11.0", |
|
|
|
"cvxopt>=1.3.0", |
|
|
|
"tqdm>=4.65.0", |
|
|
|
"scikit-learn>=0.22", |
|
|
|
"joblib>=1.2.0", |
|
|
|
"pyyaml>=6.0", |
|
|
|
"fire>=0.3.1", |
|
|
|
"lightgbm>=3.3.0", |
|
|
|
"psutil>=5.9.4", |
|
|
|
"torchvision>=0.15.1", |
|
|
|
"sqlalchemy>=2.0.21", |
|
|
|
"shortuuid>=1.0.11", |
|
|
|
"geatpy>=2.7.0", |
|
|
|
"docker>=6.1.3", |
|
|
|
"rapidfuzz>=3.4.0", |
|
|
|
"torchtext>=0.16.0", |
|
|
|
"sentence_transformers>=2.2.2", |
|
|
|
"torch-optimizer>=0.3.0", |
|
|
|
"langdetect>=1.0.9", |
|
|
|
"huggingface-hub<0.18", |
|
|
|
"portalocker>=2.0.0", |
|
|
|
# "numpy>=1.20.0", |
|
|
|
# "pandas>=0.25.1", |
|
|
|
# "scipy>=1.0.0", |
|
|
|
# "matplotlib>=3.1.3", |
|
|
|
# "torch>=1.11.0", |
|
|
|
# "cvxopt>=1.3.0", |
|
|
|
# "tqdm>=4.65.0", |
|
|
|
# "scikit-learn>=0.22", |
|
|
|
# "joblib>=1.2.0", |
|
|
|
# "pyyaml>=6.0", |
|
|
|
# "fire>=0.3.1", |
|
|
|
# "lightgbm>=3.3.0", |
|
|
|
# "psutil>=5.9.4", |
|
|
|
# "torchvision>=0.15.1", |
|
|
|
# "sqlalchemy>=2.0.21", |
|
|
|
# "shortuuid>=1.0.11", |
|
|
|
# "geatpy>=2.7.0", |
|
|
|
# "docker>=6.1.3", |
|
|
|
# "rapidfuzz>=3.4.0", |
|
|
|
# "torchtext>=0.16.0", |
|
|
|
# "sentence_transformers>=2.2.2", |
|
|
|
# "torch-optimizer>=0.3.0", |
|
|
|
# "langdetect>=1.0.9", |
|
|
|
# "huggingface-hub<0.18", |
|
|
|
# "portalocker>=2.0.0", |
|
|
|
] |
|
|
|
|
|
|
|
if get_platform() != MACOS: |
|
|
|
|