Fix version

james77777778 · May 16, 2024 · 838728b · 838728b
1 parent afc0da6
commit 838728b
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/api_gen.py b/api_gen.py
@@ -1,11 +1,13 @@
 import namex
 
+from kimm._src.version import __version__
+
 namex.generate_api_files(package="kimm", code_directory="_src")
 
 # Add version string
 
 with open("kimm/__init__.py", "r") as f:
     contents = f.read()
 with open("kimm/__init__.py", "w") as f:
-    contents += "from kimm._src.version import __version__\n"
+    contents += f'__version__ = "{__version__}"\n'
     f.write(contents)
diff --git a/kimm/__init__.py b/kimm/__init__.py
@@ -11,5 +11,6 @@
 from kimm import timm_utils
 from kimm import utils
 from kimm._src.utils.model_registry import list_models
-from kimm._src.version import __version__
 from kimm._src.version import version
+
+__version__ = "0.2.0"