setup.py

import subprocess
from setuptools import setup, find_packages
from setuptools.command.install import install

try:
    from pip._internal.req import parse_requirements
except ImportError:
    from pip.req import parse_requirements


def load_requirements(file_name):
    requirements = parse_requirements(file_name, session="test")
    try:
        return [str(item.req) for item in requirements]
    except:
        return [str(item.requirement) for item in requirements]


setup(
    name="musco-pytorch",
    version="1.0.6",
    description="MUSCO: Multi-Stage COmpression of Neural Networks",
    author="Julia Gusak, Maksym Kholiavchenko, Evgeny Ponomarev, Larisa Markeeva, Andrzej Cichocki, Ivan Oseledets",
    author_email="julgusak@gmail.com",
    url="https://github.com/musco-ai/musco-pytorch",
    download_url="https://github.com/musco-ai/musco-pytorch/archive/1.0.6.tar.gz",
    license="BSD 3-Clause",
    packages=find_packages(),
    install_requires=load_requirements("requirements.txt")
)