diff options
Diffstat (limited to 'torch_ema')
| -rw-r--r-- | torch_ema/__init__.py | 3 | ||||
| -rw-r--r-- | torch_ema/_version.py | 1 |
2 files changed, 3 insertions, 1 deletions
diff --git a/torch_ema/__init__.py b/torch_ema/__init__.py index 6cf180f..91c6bbf 100644 --- a/torch_ema/__init__.py +++ b/torch_ema/__init__.py @@ -1,3 +1,4 @@ +from ._version import __version__ from .ema import ExponentialMovingAverage -__all__ = [ExponentialMovingAverage] +__all__ = [__version__, ExponentialMovingAverage] diff --git a/torch_ema/_version.py b/torch_ema/_version.py new file mode 100644 index 0000000..6a35e85 --- /dev/null +++ b/torch_ema/_version.py @@ -0,0 +1 @@ +__version__ = "0.3" |