64 lines
1.6 KiB
Python
64 lines
1.6 KiB
Python
from codecs import open
|
|
from os import path
|
|
|
|
from setuptools import find_packages, setup
|
|
|
|
|
|
def get_long_description():
|
|
here = path.abspath(path.dirname(__file__))
|
|
|
|
with open(path.join(here, 'README.md'), encoding='utf-8') as f:
|
|
long_description = f.read()
|
|
return long_description
|
|
|
|
|
|
def get_version():
|
|
version_filepath = path.join(path.dirname(__file__), 'nyaggle', 'version.py')
|
|
with open(version_filepath) as f:
|
|
for line in f:
|
|
if line.startswith('__version__'):
|
|
return line.strip().split()[-1][1:-1]
|
|
|
|
|
|
setup(
|
|
name='nyaggle',
|
|
packages=find_packages(),
|
|
|
|
version=get_version(),
|
|
|
|
license='MIT',
|
|
|
|
install_requires=[
|
|
'category_encoders',
|
|
'matplotlib',
|
|
'more-itertools',
|
|
'numpy',
|
|
'optuna>=1.0.0',
|
|
'pandas',
|
|
'pyarrow',
|
|
'seaborn',
|
|
'scikit-learn',
|
|
'tqdm',
|
|
'transformers>=2.3.0',
|
|
],
|
|
|
|
extras_require={
|
|
'all': ['catboost>=0.17', 'lightgbm', 'xgboost', 'torch', 'mlflow']
|
|
},
|
|
|
|
author='nyanp',
|
|
author_email='Noumi.Taiga@gmail.com',
|
|
url='https://github.com/nyanp/nyaggle',
|
|
description='Code for Kaggle and Offline Competitions.',
|
|
long_description=get_long_description(),
|
|
long_description_content_type='text/markdown',
|
|
keywords='nyaggle kaggle',
|
|
classifiers=[
|
|
'License :: OSI Approved :: BSD License',
|
|
'Programming Language :: Python :: 3.8',
|
|
'Programming Language :: Python :: 3.9',
|
|
'Programming Language :: Python :: 3.10',
|
|
'Programming Language :: Python :: 3.11'
|
|
]
|
|
)
|