import os import subprocess import setuptools def git_hash(): if os.path.exists('.git'): hash = subprocess.check_output(['git', 'rev-parse', 'HEAD']).strip() return hash[:7] if (hash is not None) else None else: return None # def get_version(): from version import __version__ hash = git_hash() version_str = __version__ + '+' + hash.strip().decode('ascii') if (hash is not None) else __version__ return version_str if __name__ == '__main__': version_str = get_version() long_description = '' with open('README.md') as readme: long_description = readme.read() setuptools.setup( name = 'pytorch-jacinto-ai', version = get_version(), description = 'Jacinto AI DevKit (PyTorch Version)', long_description = long_description, long_description_content_type = 'text/markdown', url = 'http://git.ti.com/jacinto-ai/pytorch-jacinto-ai', author = 'Jacinto AI DevKit Team', author_email = 'jacinto-ai@list.ti.com', classifiers = [ 'Development Status :: 4 - Beta' 'Programming Language :: Pyuthon :: 3.7' ], keywords = 'artifical intelligence, deep learning, image classification, object detection, semantic segmentation, quantization', #packages=[setuptools.find_packages(exclude=[])] #do not have sub packages python_requires = '>=3.7', install_requires = [], package_dir = {'':'modules'}, package_data = {'':['*.txt']}, project_urls = { 'Source': 'http://git.ti.com/jacinto-ai/pytorch-jacinto-ai', 'Bug Reports': 'http://e2e.ti.com/search?q=J7&category=forum', }, )