[FBGEMM][v0.5.0-rc3] Fix using package version and missing bash path

davidberard98 · Sep 18, 2023 · 52a96be · 52a96be
1 parent 87d2327
commit 52a96be
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/.github/scripts/setup_env.bash b/.github/scripts/setup_env.bash
@@ -22,6 +22,8 @@
 # shellcheck disable=SC1091,SC2128
 . "$( dirname -- "$BASH_SOURCE"; )/fbgemm_gpu_docs.bash"
 # shellcheck disable=SC1091,SC2128
+. "$( dirname -- "$BASH_SOURCE"; )/fbgemm_gpu_install.bash"
+# shellcheck disable=SC1091,SC2128
 . "$( dirname -- "$BASH_SOURCE"; )/fbgemm_gpu_lint.bash"
 # shellcheck disable=SC1091,SC2128
 . "$( dirname -- "$BASH_SOURCE"; )/fbgemm_gpu_test.bash"

diff --git a/fbgemm_gpu/setup.py b/fbgemm_gpu/setup.py
@@ -33,7 +33,7 @@ def generate_package_version(package_name: str, version_variant: str):
         today = date.today()
         version = f"{today.year}.{today.month}.{today.day}"
 
-    elif "test" in package_name:
+    elif ("test" in package_name) and (not "BUILD_FROM_NOVA" in os.environ):
         # Use date stamp for nightly versions
         print("[SETUP.PY] Package is for TEST: using random number for the versioning")
         version = (f"0.0.{random.randint(0, 1000)}",)