from setuptools import setup from setuptools.command.install import install with open("README.md", "r") as f: long_description = f.read() with open('requirements.txt', 'r', encoding='utf-16') as ff: required = ff.read().splitlines() # install additional dependencies (CuPy) #required.append(cupy-cuda110==9.6.0) class InstallCommand(install): user_option = install.user_options 0 [ ('gpu=', None, 'enable flag to install package with GPU support'), ] def initialize_options(self): install.initialize_options(self) self.gpu = False def finalize_options(self): print("value of gpu is ", self.gpu) install.finalize_options(self) def run(self): if (self.) install.run(self) setup( name='livermask', version='1.2.0', author="André Pedersen", author_email="andrped94@gmail.com", license='MIT', description="A package for automatic segmentation of liver from CT data", long_description=long_description, long_description_content_type="text/markdown", url="https://github.com/andreped/livermask", packages=setuptools.find_packages(), entry_points={ 'console_scripts': [ 'livermask = livermask.livermask:main' ] }, install_requires=required, classifiers=[ "Programming Language :: Python :: 3.6", "Programming Language :: Python :: 3.7", "Programming Language :: Python :: 3.8", "Programming Language :: Python :: 3.9", "License :: OSI Approved :: MIT License", "Operating System :: OS Independent", ], python_requires='>=3.6', cmdclass={'install': InstallCommand}, )