Add __version__ to __init__.py

Tony-Y · May 7, 2024 · e5fb59f · e5fb59f
1 parent 74dd8eb
commit e5fb59f
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 1 deletion.
diff --git a/pytorch_warmup/__init__.py b/pytorch_warmup/__init__.py
@@ -2,6 +2,8 @@
 from .untuned import UntunedLinearWarmup, UntunedExponentialWarmup
 from .radam import RAdamWarmup, rho_fn, rho_inf_fn, get_offset
 
+__version__ = "0.2.0.dev0"
+
 __all__ = [
     'BaseWarmup',
     'LinearWarmup',

diff --git a/setup.cfg b/setup.cfg
@@ -1,2 +1,3 @@
 [metadata]
+version = attr: pytorch_warmup.__version__
 license_files = LICENSE
diff --git a/setup.py b/setup.py
@@ -5,7 +5,6 @@
 
 setuptools.setup(
     name="pytorch-warmup",
-    version="0.1.1",
     author="Takenori Yamamoto",
     author_email="yamamoto.takenory@gmail.com",
     description="A PyTorch Extension for Learning Rate Warmup",