Skip to content

Commit

Permalink
Refactor and fix lost requirements.txt file
Browse files Browse the repository at this point in the history
  • Loading branch information
nok committed Aug 21, 2017
1 parent c6d6a65 commit 2935828
Show file tree
Hide file tree
Showing 3 changed files with 21 additions and 16 deletions.
1 change: 1 addition & 0 deletions MANIFEST.in
@@ -1 +1,2 @@
include requirements.txt
recursive-include sklearn_porter *.txt *.py
34 changes: 19 additions & 15 deletions setup.py
@@ -1,29 +1,33 @@
# -*- coding: utf-8 -*-

import os
from setuptools import setup, find_packages
from setuptools import setup
from setuptools import find_packages


path = os.path.abspath(os.path.dirname(__file__))
def read_version():
src_dir = os.path.abspath(os.path.dirname(__file__))
ver_file = os.path.join(src_dir, 'sklearn_porter', '__version__.txt')
version = open(ver_file, 'r').readlines().pop()
if isinstance(version, bytes):
version = version.decode('utf-8')
version = str(version).strip()
return version

# Version:
version_path = str(os.path.join(path, 'sklearn_porter', '__version__.txt'))
version = open(version_path).readlines().pop()
if isinstance(version, bytes):
version = version.decode('utf-8')
version = str(version).strip()

# Requirements:
requirements_path = os.path.join(path, 'requirements.txt')
with open(requirements_path) as f:
all_reqs = f.read().split('\n')
requirements = [x.strip() for x in all_reqs if 'git+' not in x]
def parse_requirements():
src_dir = os.path.abspath(os.path.dirname(__file__))
req_file = os.path.join(src_dir, 'requirements.txt')
reqs = open(req_file, 'r').read().strip().split('\n')
reqs = [req.strip() for req in reqs if 'git+' not in req]
return reqs


setup(
name='sklearn-porter',
packages=find_packages(exclude=["tests.*", "tests"]),
include_package_data=True,
version=version,
version=read_version(),
description='Transpile trained scikit-learn models to C, Java, JavaScript and others.',
author='Darius Morawiec',
author_email='ping@nok.onl',
Expand All @@ -38,7 +42,7 @@
'Programming Language :: Python :: 3.5',
'Programming Language :: Python :: 3.6',
],
install_requires=requirements,
install_requires=parse_requirements(),
keywords=['sklearn', 'scikit-learn'],
license='MIT',
)
2 changes: 1 addition & 1 deletion sklearn_porter/__version__.txt
@@ -1 +1 @@
0.5.0
0.5.1

0 comments on commit 2935828

Please sign in to comment.