diff --git a/learnware/reuse/__init__.py b/learnware/reuse/__init__.py index 0ea82c5..8e9621b 100644 --- a/learnware/reuse/__init__.py +++ b/learnware/reuse/__init__.py @@ -1,3 +1,5 @@ from .ensemble_pruning import EnsemblePruningReuser from .averaging import AveragingReuser from .job_selector import JobSelectorReuser +from .feature_augment_reuser import FeatureAugmentReuser +from .hetero_reuser import HeteroMapTableReuser \ No newline at end of file diff --git a/setup.py b/setup.py index 67f7254..b37dfab 100644 --- a/setup.py +++ b/setup.py @@ -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: