forked from apple/coremltools
-
Notifications
You must be signed in to change notification settings - Fork 0
/
setup.py
59 lines (54 loc) · 1.88 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
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
#!/usr/bin/env python
import os
from setuptools import setup
README = os.path.join(os.getcwd(), "README.rst")
with open(README) as f:
long_description = f.read()
setup(name='coremltools',
version='0.7',
description='Community Tools for CoreML',
long_description=long_description,
author='Apple Inc.',
author_email='[email protected]',
url='',
packages=[
'coremltools',
'coremltools._deps',
'coremltools.converters',
'coremltools.converters.caffe',
'coremltools.converters.sklearn',
'coremltools.converters.xgboost',
'coremltools.converters.libsvm',
'coremltools.converters.keras',
'coremltools.graph_visualization',
'coremltools.models',
'coremltools.proto',
'coremltools._scripts'
],
package_data={'': ['LICENSE.txt', 'README.rst', 'libcaffeconverter.so', 'libcoremlpython.so'],
'coremltools': ['graph_visualization/__init__.py',
'graph_visualization/app.js',
'graph_visualization/index.html',
'graph_visualization/style.css',
'graph_visualization/assets/*',
'graph_visualization/icons/*']
},
install_requires=[
'numpy >= 1.6.2',
'protobuf >= 3.1.0',
'six==1.10.0'
],
entry_points = {
'console_scripts': ['coremlconverter = coremltools:_main']
},
classifiers=[
'Development Status :: 4 - Beta',
'Intended Audience :: End Users/Desktop',
'Intended Audience :: Developers',
'Operating System :: MacOS :: MacOS X',
'Programming Language :: Python :: 2.7',
'Topic :: Scientific/Engineering',
'Topic :: Software Development'
],
license='BSD'
)