From 0cc9341605373ec318fc7a5c5ae423531db4d49b Mon Sep 17 00:00:00 2001
From: Isaac Yang <isaacy@nvidia.com>
Date: Thu, 13 Jun 2024 11:24:40 -0700
Subject: [PATCH] Update setup.py for base version and nightly build

---
 setup.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index 19825a8e4e..fc51b6d0a8 100644
--- a/setup.py
+++ b/setup.py
@@ -24,12 +24,16 @@
 # read the contents of your README file
 
 versions = versioneer.get_versions()
+base_version = os.environ.get("NVFL_BASE_VERSION")
 if versions["error"]:
     today = datetime.date.today().timetuple()
     year = today[0] % 1000
     month = today[1]
     day = today[2]
-    version = f"2.3.0.dev{year:02d}{month:02d}{day:02d}"
+    if base_version:
+        version = f"{base_version}.dev{year:02d}{month:02d}{day:02d}"
+    else:
+        version = f"2.5.0.dev{year:02d}{month:02d}{day:02d}"
 else:
     version = versions["version"]