fix version thing

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 = []