fixup! [Utils] Add deprecation handling for kwargs with deprecate_kwarg decorator (#455)

Author: zhiyuan1i

fla/__init__.py

@@ -1,5 +1,6 @@
 # -*- coding: utf-8 -*-
 
+from fla._version import __version__
 from fla.layers import (
     ABCAttention,
     Attention,
@@ -68,6 +69,7 @@
 )
 
 __all__ = [
+    '__version__',
     'ABCAttention', 'ABCForCausalLM', 'ABCModel',
     'Attention', 'TransformerForCausalLM', 'TransformerModel',
     'BasedLinearAttention',
@@ -91,5 +93,3 @@
     'RWKV6Attention', 'RWKV6ForCausalLM', 'RWKV6Model',
     'RWKV7Attention', 'RWKV7ForCausalLM', 'RWKV7Model',
 ]
-
-__version__ = '0.3.0'

fla/_version.py

@@ -0,0 +1,3 @@
+# -*- coding: utf-8 -*-
+
+__version__ = '0.3.0'

fla/utils.py

@@ -15,7 +15,7 @@
 import triton
 from packaging import version
 
-from fla import __version__
+from fla._version import __version__
 
 logger = logging.getLogger(__name__)
 

setup.py

@@ -12,7 +12,7 @@
 
 
 def get_package_version():
-    with open(Path(os.path.dirname(os.path.abspath(__file__))) / 'fla' / '__init__.py') as f:
+    with open(Path(os.path.dirname(os.path.abspath(__file__))) / 'fla' / '_version.py') as f:
         version_match = re.search(r"^__version__\s*=\s*(.*)$", f.read(), re.MULTILINE)
     return ast.literal_eval(version_match.group(1))