from setuptools import setup, find_packages from pathlib import Path # see https://packaging.python.org/guides/single-sourcing-package-version/ version_dict = {} with open(Path(__file__).parents[0] / "torch_runstats/_version.py") as fp: exec(fp.read(), version_dict) version = version_dict["__version__"] del version_dict url = 'https://github.com/fadel/pytorch_ema' download_url = '{}/archive/{}.tar.gz'.format(url, __version__) install_requires = ["torch"] setup_requires = [] tests_require = [] setup( name='torch_ema', version=version, description='PyTorch library for computing moving averages of model parameters.', author='Samuel G. Fadel', author_email='samuelfadel@gmail.com', url=url, download_url=download_url, keywords=['pytorch', 'parameters', 'deep-learning'], install_requires=install_requires, setup_requires=setup_requires, tests_require=tests_require, packages=find_packages(), )