Fix issues with version

keras_nlp/__init__.py

@@ -26,4 +26,5 @@
 from keras_nlp import samplers
 from keras_nlp import tokenizers
 from keras_nlp import utils
-from keras_nlp.version import __version__
+from keras_nlp.version_utils import __version__
+from keras_nlp.version_utils import version

pip_build.py

@@ -64,10 +64,19 @@ def export_version_string(version, is_nightly=False):
             f.write(setup_contents)
 
     # Make sure to export the __version__ string
+    with open(os.path.join(package, "src", "version_utils.py")) as f:
+        version_contents = list(f.readlines())
+    with open(os.path.join(package, "src", "version_utils.py"), "w") as f:
+        for line in version_contents:
+            if line.startswith("__version__"):
+                f.write(f'__version__ = "{version}"')
+            else:
+                f.write(line)
     with open(os.path.join(package, "__init__.py")) as f:
         init_contents = f.read()
     with open(os.path.join(package, "__init__.py"), "w") as f:
-        f.write(init_contents + "\n\n" + f'__version__ = "{version}"\n')
+        f.write(init_contents + "\n\n")
+        f.write("from keras_nlp.src import __version__\n")
 
 
 def build(root_path, is_nightly=False):

setup.py

@@ -37,7 +37,10 @@ def get_version(rel_path):
 
 HERE = pathlib.Path(__file__).parent
 README = (HERE / "README.md").read_text()
-VERSION = get_version("keras_nlp/version.py")
+if os.path.exists("keras_nlp/version_utils.py"):
+    VERSION = get_version("keras_nlp/version_utils.py")
+else:
+    VERSION = get_version("keras_nlp/src/version_utils.py")
 
 setup(
     name="keras-nlp",