diff options
Diffstat (limited to 'torch_ema/__init__.py')
| -rw-r--r-- | torch_ema/__init__.py | 3 |
1 files changed, 2 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] |