aboutsummaryrefslogtreecommitdiff
path: root/setup.py
diff options
context:
space:
mode:
Diffstat (limited to 'setup.py')
-rw-r--r--setup.py4
1 files changed, 2 insertions, 2 deletions
diff --git a/setup.py b/setup.py
index 49b8704..0fdff25 100644
--- a/setup.py
+++ b/setup.py
@@ -3,13 +3,13 @@ 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:
+with open(Path(__file__).parents[0] / "torch_ema/_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__)
+download_url = '{}/archive/{}.tar.gz'.format(url, version)
install_requires = ["torch"]
setup_requires = []