setup.py

import os

from setuptools import setup, Command


with open(os.path.join(os.path.dirname(__file__), 'requirements.txt')) as f:
    required = f.read().splitlines()


class PyTest(Command):
    user_options = []

    def initialize_options(self):
        pass

    def finalize_options(self):
        pass

    def run(self):
        import subprocess
        import sys
        errno = subprocess.call([sys.executable, 'runtests.py'])
        raise SystemExit(errno)

setup(
    name='sklearn-pmml',
    version='0.0.1',
    packages=['sklearn_pmml', 'sklearn_pmml.convert', 'sklearn_pmml.convert.test'],
    install_requires=required,
    cmdclass={'test': PyTest},
    url='https://github.com/alex-pirozhenko/sklearn-pmml',
    license='MIT',
    author='Alex Pirozhenko',
    author_email='apirozhenko@pulsepoint.com',
    description='A library that allows serialization of SciKit-Learn estimators into PMML'
)