forked from scikit-learn-contrib/skglm
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathsetup.py
35 lines (31 loc) · 1.22 KB
/
setup.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
import os
from setuptools import setup, find_packages
version = None
with open(os.path.join('skglm', '__init__.py'), 'r') as fid:
for line in (line.strip() for line in fid):
if line.startswith('__version__'):
version = line.split('=')[1].strip().strip('\'')
break
if version is None:
raise RuntimeError('Could not determine version')
DISTNAME = 'skglm'
DESCRIPTION = 'Fast and modular scikit-learn replacement for generalized linear models'
MAINTAINER = 'Mathurin Massias'
MAINTAINER_EMAIL = 'mathurin.massias@gmail.com'
LICENSE = 'BSD (3-clause)'
DOWNLOAD_URL = 'https://github.com/scikit-learn-contrib/skglm.git'
VERSION = version
URL = 'https://contrib.scikit-learn.org/skglm'
setup(name=DISTNAME,
version=version,
description=DESCRIPTION,
long_description=open('README.rst').read(),
maintainer=MAINTAINER,
maintainer_email=MAINTAINER_EMAIL,
url=URL,
download_url=DOWNLOAD_URL,
packages=find_packages(),
install_requires=['libsvmdata>=0.2', 'numpy>=1.12', 'numba',
'seaborn>=0.7',
'joblib', 'scipy>=0.18.0', 'matplotlib>=2.0.0',
'scikit-learn>=1.0', 'pandas', 'ipython'],)