fix: Generate version number string (#467)

Signed-off-by: oliver könig <[email protected]>
NVIDIA · Jan 3, 2025 · 7dfb21a · 7dfb21a
1 parent 98fb0bc
commit 7dfb21a
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/nemo_curator/package_info.py b/nemo_curator/package_info.py
@@ -23,7 +23,13 @@
 VERSION = (MAJOR, MINOR, PATCH, PRE_RELEASE, DEV)
 
 __shortversion__ = ".".join(map(str, VERSION[:3]))
-__version__ = __shortversion__ + VERSION[3] + "." + ".".join(VERSION[4:])
+__version__ = __shortversion__
+
+if VERSION[3] != "":
+    __version__ = __version__ + VERSION[3]
+
+if VERSION[4] != "":
+    __version__ = __version__ + "." + ".".join(VERSION[4:])
 
 __package_name__ = "nemo_curator"
 __contact_names__ = "NVIDIA"