Commit 770fdc16 authored by rusty1s's avatar rusty1s
Browse files

better version impl

parent 7d13d713
from os import path as osp from os import path as osp
from setuptools import setup, find_packages from setuptools import setup, find_packages
filename = osp.join(osp.dirname(osp.abspath('__file__')), 'VERSION')
with open(filename, 'r') as f:
version = f.read().strip()
import build # noqa import build # noqa
install_requires = ['cffi'] install_requires = ['cffi']
...@@ -10,7 +14,7 @@ docs_require = ['Sphinx', 'sphinx_rtd_theme'] ...@@ -10,7 +14,7 @@ docs_require = ['Sphinx', 'sphinx_rtd_theme']
setup( setup(
name='torch_scatter', name='torch_scatter',
version='0.1.0', version=version,
description='PyTorch extension for various scatter methods', description='PyTorch extension for various scatter methods',
url='https://github.com/rusty1s/pytorch_scatter', url='https://github.com/rusty1s/pytorch_scatter',
author='Matthias Fey', author='Matthias Fey',
......
from os import path as osp
from .functions.add import scatter_add_, scatter_add from .functions.add import scatter_add_, scatter_add
from .functions.sub import scatter_sub_, scatter_sub from .functions.sub import scatter_sub_, scatter_sub
from .functions.mul import scatter_mul_, scatter_mul from .functions.mul import scatter_mul_, scatter_mul
...@@ -6,7 +8,9 @@ from .functions.mean import scatter_mean_, scatter_mean ...@@ -6,7 +8,9 @@ from .functions.mean import scatter_mean_, scatter_mean
from .functions.max import scatter_max_, scatter_max from .functions.max import scatter_max_, scatter_max
from .functions.min import scatter_min_, scatter_min from .functions.min import scatter_min_, scatter_min
from .version import __version__ filename = osp.join(osp.dirname(osp.abspath('__file__')), 'VERSION')
with open(filename, 'r') as f:
__version__ = f.read().strip()
__all__ = [ __all__ = [
'scatter_add_', 'scatter_add', 'scatter_sub_', 'scatter_sub', 'scatter_add_', 'scatter_add', 'scatter_sub_', 'scatter_sub',
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment