|
4 | 4 | import os |
5 | 5 | rootdir = os.path.dirname(os.path.realpath(__file__)) |
6 | 6 |
|
7 | | -def readme(): |
8 | | - with open('README.md') as f: |
9 | | - return f.read() |
| 7 | +version = "0.1.1" |
10 | 8 |
|
11 | 9 | setup(name='pytorch_block_sparse', |
12 | | - version='v0.1', |
13 | | - description='pytorch_block_sparse is a python package for fast block sparse matrices computation, drop in replacement for torch.nn.Linear .', |
14 | | - long_description=readme(), |
| 10 | + version=version, |
| 11 | + description='PyTorch extension for fast block sparse matrices computation, drop in replacement for torch.nn.Linear.', |
| 12 | + long_description="pytorch_block_sparse is a PyTorch extension for fast block sparse matrices computation, drop in replacement for torch.nn.Linear", |
15 | 13 | classifiers=[ |
16 | 14 | 'Development Status :: 4 - Beta', |
17 | | - 'License :: OSI Approved :: BSD 3-Clause "New" or "Revised" License', |
| 15 | + 'License :: OSI Approved :: BSD License', |
18 | 16 | 'Programming Language :: Python :: 3.0', |
19 | 17 | ], |
20 | | - keywords='', |
21 | | - url='', |
22 | | - author='', |
23 | | - author_email='', |
24 | | - download_url='https://github.com/huggingface/pytorch_block_sparse/archive/v0.1.tar.gz', |
| 18 | + keywords='PyTorch,sparse,matrices,machine learning', |
| 19 | + url='https://github.com/huggingface/pytorch_block_sparse', |
| 20 | + author='François Lagunas', |
| 21 | + author_email='francois.lagunas@m4x.org', |
| 22 | + download_url=f'https://test.pypi.org/project/pytorch-block-sparse/{version}/', |
25 | 23 | license='BSD 3-Clause "New" or "Revised" License', |
26 | 24 | packages=['pytorch_block_sparse'], |
27 | 25 | install_requires=[], |
|
0 commit comments