remove dependency to catboost/lightgbm/xgboost
parent
7b390b976c
commit
50007e0b8b
|
@ -21,7 +21,7 @@ jobs:
|
|||
run: |
|
||||
python -m pip install --upgrade pip
|
||||
pip install -r requirements.txt
|
||||
pip install torch mlflow
|
||||
pip install torch mlflow catboost lightgbm xgboost
|
||||
- name: Install MeCab
|
||||
run: |
|
||||
sudo apt install mecab libmecab-dev mecab-ipadic-utf8
|
||||
|
|
|
@ -17,7 +17,7 @@ jobs:
|
|||
run: |
|
||||
python -m pip install --upgrade pip
|
||||
pip install -r requirements.txt
|
||||
pip install torch mlflow
|
||||
pip install torch mlflow catboost lightgbm xgboost
|
||||
pip install setuptools==42.0.2
|
||||
pip install wheel twine
|
||||
- name: Build and publish
|
||||
|
|
|
@ -47,6 +47,7 @@ def requires_lightgbm():
|
|||
try:
|
||||
import catboost
|
||||
_has_catboost = True
|
||||
# TODO check catboost version >= 0.17
|
||||
except ImportError:
|
||||
_has_catboost = False
|
||||
|
||||
|
|
|
@ -1,6 +1,4 @@
|
|||
catboost>=0.17
|
||||
category_encoders
|
||||
lightgbm
|
||||
matplotlib
|
||||
more-itertools
|
||||
numpy
|
||||
|
@ -11,4 +9,3 @@ seaborn
|
|||
sklearn
|
||||
tqdm
|
||||
transformers
|
||||
xgboost
|
||||
|
|
5
setup.py
5
setup.py
|
@ -28,9 +28,7 @@ setup(
|
|||
license='MIT',
|
||||
|
||||
install_requires=[
|
||||
'catboost>=0.17',
|
||||
'category_encoders',
|
||||
'lightgbm',
|
||||
'matplotlib',
|
||||
'more-itertools',
|
||||
'numpy',
|
||||
|
@ -40,8 +38,7 @@ setup(
|
|||
'seaborn',
|
||||
'sklearn',
|
||||
'tqdm',
|
||||
'transformers>=2.3.0',
|
||||
'xgboost'
|
||||
'transformers>=2.3.0'
|
||||
],
|
||||
author='nyanp',
|
||||
author_email='Noumi.Taiga@gmail.com',
|
||||
|
|
Loading…
Reference in New Issue