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

better version impl

parent 7d13d713
from os import path as osp
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
install_requires = ['cffi']
......@@ -10,7 +14,7 @@ docs_require = ['Sphinx', 'sphinx_rtd_theme']
setup(
name='torch_scatter',
version='0.1.0',
version=version,
description='PyTorch extension for various scatter methods',
url='https://github.com/rusty1s/pytorch_scatter',
author='Matthias Fey',
......
from os import path as osp
from .functions.add import scatter_add_, scatter_add
from .functions.sub import scatter_sub_, scatter_sub
from .functions.mul import scatter_mul_, scatter_mul
......@@ -6,7 +8,9 @@ from .functions.mean import scatter_mean_, scatter_mean
from .functions.max import scatter_max_, scatter_max
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__ = [
'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