Skip to content

Commit 6a80b3f

Browse files
committed
Added setup
1 parent c88fe5b commit 6a80b3f

File tree

1 file changed

+30
-0
lines changed

1 file changed

+30
-0
lines changed

setup.py

+30
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,30 @@
1+
#!/usr/bin/env python
2+
3+
import os
4+
import graph_ensemble
5+
6+
7+
DEPENDENCIES = ['numpy', 'sklearn']
8+
9+
metadata = dict()
10+
11+
try:
12+
from setuptools import setup, find_packages
13+
packages = find_packages()
14+
metadata['install_requires'] = DEPENDENCIES
15+
except ImportError:
16+
from distutils.core import setup
17+
from pkgutil import walk_packages
18+
packages = [package for _, package, ispkg in walk_packages('.') if ispkg]
19+
metadata['requires'] = DEPENDENCIES
20+
21+
22+
metadata['name'] = 'GraphEnsemble'
23+
metadata['description'] = 'A framework for creating arbitrary graphs of machine learning models'
24+
metadata['version'] = graph_ensemble.__version__
25+
metadata['license'] = 'BSD 3-Clause License'
26+
metadata['url'] = 'https://github.com/YotamFY/GraphEnsemble'
27+
metadata['packages'] = packages
28+
29+
setup(**metadata)
30+

0 commit comments

Comments
 (0)