-
Notifications
You must be signed in to change notification settings - Fork 1
/
setup.py
29 lines (28 loc) · 896 Bytes
/
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
from setuptools import setup
import setuptools
setup(
name='cords',
version='1.0.0',
author='Krishnateja Killamsetty, Dheeraj Bhat, Rishabh Iyer',
author_email='[email protected]',
#packages=['cords', 'cords/selectionstrategies', 'cords/utils'],
url='https:/decile-team/cords',
license='LICENSE.txt',
packages=setuptools.find_packages(),
description='cords is a package for data subset selection for efficient and robust machine learning.',
install_requires=[
"numpy >= 1.14.2",
"scipy >= 1.0.0",
"numba >= 0.43.0",
"tqdm >= 4.24.0",
"torch >= 1.8.0",
"apricot-select >= 0.6.0",
"sphinxcontrib-napoleon",
"sphinxcontrib-bibtex",
"sphinx-rtd-theme",
"scikit-learn",
"torchvision >= 0.5.0",
"matplotlib",
"ray[tune]"
],
)