forked from alvinhenrick/spark_ml
-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy pathsetup.py
executable file
·34 lines (26 loc) · 1007 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
30
31
32
33
34
from os import path, getcwd
from setuptools import setup, find_packages
package_name = 'spark_ml'
try:
with open(path.join(getcwd(), 'VERSION')) as version_file:
version = version_file.read().strip()
except IOError:
raise
def parse_requirements(file):
with open(file, "r") as fs:
return [r for r in fs.read().splitlines() if
(len(r.strip()) > 0 and not r.strip().startswith("#") and not r.strip().startswith("--"))]
requirements = parse_requirements('requirements.txt')
test_requirements = parse_requirements('requirements-dev.txt')
setup(name=package_name,
version=version,
license='',
description='Project to run quick spark ml',
author='Alvin Henrick',
author_email='[email protected]',
url='https://github.com/alvinhenrick/spark_ml',
packages=find_packages(exclude=['tests']),
install_requires=requirements,
tests_require=test_requirements,
include_package_data=True,
zip_safe=False)