1 Star 0 Fork 0

/tensorboardX

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
克隆/下载
setup.py 2.75 KB
一键复制 编辑 原始数据 按行查看 历史
Tzu-Wei Huang 提交于 2021-04-03 23:54 . Release 2.2 (#622)
#!/usr/bin/env python
# -*- coding: utf-8 -*-
import subprocess
import os
import sys
from setuptools import setup, find_packages
from setuptools.command.develop import develop
from setuptools.command.install import install
# Dynamically compile protos
def compileProtoBuf():
res = subprocess.call(['bash', './compile.sh'])
assert res == 0, 'cannot compile protobuf'
class PostDevelopCommand(develop):
"""Post-installation for development mode."""
def run(self):
compileProtoBuf()
develop.run(self)
class PostInstallCommand(install):
"""Post-installation for installation mode."""
def run(self):
compileProtoBuf()
import os
os.system("pip install protobuf numpy")
install.run(self)
with open('HISTORY.rst') as history_file:
history = history_file.read()
preparing_PyPI_package = 'sdist' in sys.argv or 'bdist_wheel' in sys.argv
version_git = version = '2.2'
if not preparing_PyPI_package:
if os.path.exists('.git'):
sha = subprocess.check_output(['git', 'rev-parse', 'HEAD']).decode('ascii').strip()
version_git = version_git + '+' + sha[:7]
with open('tensorboardX/__init__.py', 'a') as f:
f.write('\n__version__ = "{}"\n'.format(version_git))
requirements = [
'numpy',
'protobuf >= 3.8.0',
]
test_requirements = [
'future'
'pytest',
'matplotlib',
'crc32c',
]
setup(
name='tensorboardX',
version=version_git,
description='TensorBoardX lets you watch Tensors Flow without Tensorflow',
long_description=history,
author='Tzu-Wei Huang',
author_email='huang.dexter@gmail.com',
url='https://github.com/lanpa/tensorboardX',
packages=['tensorboardX'],
include_package_data=True,
install_requires=requirements,
license='MIT license',
zip_safe=False,
classifiers=[
'Development Status :: 4 - Beta',
'Intended Audience :: Developers',
'License :: OSI Approved :: MIT License',
'Natural Language :: English',
'Programming Language :: Python :: 3',
'Programming Language :: Python :: 3.6',
'Programming Language :: Python :: 3.7',
'Programming Language :: Python :: 3.8',
],
cmdclass={
'develop': PostDevelopCommand,
'install': PostInstallCommand,
},
test_suite='tests',
tests_require=test_requirements
)
# checklist: update History.rst readme.md
# update the version number in this file (setup.py).
# python setup.py sdist bdist_wheel --universal
# check the generated tar.gz file
# (1. The version number is correct. 2. no *.pyc __pycache__ files)
# git checkout -b "release x.x"
# git add setup.py History.rst readme.md
# git commit -m 'prepare for release' (skip tensorboardX/__init__.py)
# add tag
# twine upload dist/*
# git push
Loading...
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
1
https://gitee.com/guojie1993/tensorboardX.git
git@gitee.com:guojie1993/tensorboardX.git
guojie1993
tensorboardX
tensorboardX
master

搜索帮助