test new nightly

LiteRT-LM-PiperOrigin-RevId: 890579357

Author: whhone

.github/workflows/nightly-linux-arm64.yml

@@ -2,6 +2,9 @@ name: "Nightly-Linux-Arm64"
 
 on:
   workflow_dispatch:
+  pull_request:
+    branches:
+    - main
   schedule:
     - cron: '0 21 * * *'
       timezone: 'America/Los_Angeles'
@@ -14,6 +17,9 @@ jobs:
       matrix:
         python-version: ["3.10", "3.11", "3.12", "3.13"]
     steps:
+      # - name: Install required packages
+      #   run: sudo apt-get update && sudo apt-get install -y git-lfs patchelf
+
       - name: Checkout code.
         uses: actions/checkout@v4
         with:
@@ -22,7 +28,7 @@ jobs:
       - name: Build Python Wheel
         run: |
           PYTHON_VERSION=${{ matrix.python-version }}
-          DATE=$(date +'%Y%m%d')
+          DATE=$(TZ=America/Los_Angeles date +'%Y%m%d2')
           bazel build \
             --repo_env=HERMETIC_PYTHON_VERSION=${PYTHON_VERSION} \
             --@rules_python//python/config_settings:python_version=${PYTHON_VERSION} \
@@ -31,7 +37,7 @@ jobs:
             --define=litert_link_capi_so=true \
             --define=resolve_symbols_in_exec=false \
             --config=linux_arm64 \
-            -c opt //python/litert_lm:litert_lm_wheel
+            -c opt //python/litert_lm:wheel
 
       - name: Install uv
         uses: astral-sh/setup-uv@v5

.github/workflows/nightly-linux-x64.yml

@@ -2,6 +2,9 @@ name: "Nightly-Linux-x64"
 
 on:
   workflow_dispatch:
+  pull_request:
+    branches:
+    - main
   schedule:
     - cron: '0 21 * * *'
       timezone: 'America/Los_Angeles'
@@ -25,7 +28,7 @@ jobs:
         # is used to maintain compatibility with Google Colab environments.
         run: |
           PYTHON_VERSION=${{ matrix.python-version }}
-          DATE=$(date +'%Y%m%d')
+          DATE=$(TZ=America/Los_Angeles date +'%Y%m%d2')
           bazel build \
             --repo_env=HERMETIC_PYTHON_VERSION=${PYTHON_VERSION} \
             --@rules_python//python/config_settings:python_version=${PYTHON_VERSION} \
@@ -34,7 +37,7 @@ jobs:
             --define=litert_link_capi_so=true \
             --define=resolve_symbols_in_exec=false \
             --define=xnn_enable_avxvnniint8=false \
-            -c opt //python/litert_lm:litert_lm_wheel
+            -c opt //python/litert_lm:wheel
 
       - name: Install uv
         uses: astral-sh/setup-uv@v5
@@ -63,12 +66,12 @@ jobs:
       - name: Build CLI Python Wheel
         if: matrix.python-version == '3.13'
         run: |
-          DATE=$(date +'%Y%m%d')
+          DATE=$(TZ=America/Los_Angeles date +'%Y%m%d2')
           bazel build \
             --define=PYPI_NIGHTLY=1 \
             --define=PYPI_NIGHTLY_DATE=${DATE} \
             --define=xnn_enable_avxvnniint8=false \
-            -c opt //python/litert_lm_cli:litert_lm_cli_wheel
+            -c opt //python/litert_lm_cli:wheel
 
       - name: Test CLI Python Wheel
         if: matrix.python-version == '3.13'

.github/workflows/nightly-mac-arm64.yml

@@ -2,6 +2,9 @@ name: "Nightly-Mac-Arm64"
 
 on:
   workflow_dispatch:
+  pull_request:
+    branches:
+    - main
   schedule:
     - cron: '0 21 * * *'
       timezone: 'America/Los_Angeles'
@@ -22,13 +25,13 @@ jobs:
       - name: Build Python Wheel
         run: |
           PYTHON_VERSION=${{ matrix.python-version }}
-          DATE=$(date +'%Y%m%d')
+          DATE=$(TZ=America/Los_Angeles date +'%Y%m%d2')
           bazel build --config=macos_arm64 \
             --repo_env=HERMETIC_PYTHON_VERSION=${PYTHON_VERSION} \
             --@rules_python//python/config_settings:python_version=${PYTHON_VERSION} \
             --define=PYPI_NIGHTLY=1 \
             --define=PYPI_NIGHTLY_DATE=${DATE} \
-            -c opt //python/litert_lm:litert_lm_wheel
+            -c opt //python/litert_lm:wheel
 
       - name: Install uv
         uses: astral-sh/setup-uv@v5

python/litert_lm/BUILD

@@ -192,7 +192,7 @@ genrule(
 )
 
 py_package(
-    name = "litert_lm_pkg",
+    name = "package",
     # Explicitly specify packages to avoid including transitively linked shared
     # libraries (like duplicate .so files in _solib_*/...) in the wheel.
     packages = ["python.litert_lm"],
@@ -204,7 +204,7 @@ py_package(
 # $ bazel build \
 #     --repo_env=HERMETIC_PYTHON_VERSION=${PYTHON_VERSION} \
 #     --@rules_python//python/config_settings:python_version=${PYTHON_VERSION} \
-#     -c opt //python/litert_lm:litert_lm_wheel
+#     -c opt //python/litert_lm:wheel
 #
 # To build nightly wheel:
 # $ PYTHON_VERSION=3.10
@@ -214,9 +214,9 @@ py_package(
 #     --@rules_python//python/config_settings:python_version=${PYTHON_VERSION} \
 #     --define=PYPI_NIGHTLY=1 \
 #     --define=PYPI_NIGHTLY_DATE=${DATE} \
-#     -c opt //python/litert_lm:litert_lm_wheel
+#     -c opt //python/litert_lm:wheel
 py_wheel(
-    name = "litert_lm_wheel",
+    name = "wheel",
     abi = select({
         "@rules_python//python/config_settings:is_python_3.10": "cp310",
         "@rules_python//python/config_settings:is_python_3.11": "cp311",
@@ -225,8 +225,8 @@ py_wheel(
         "//conditions:default": "none",
     }),
     distribution = select({
-        ":is_nightly": "litert-lm-nightly",
-        "//conditions:default": "litert-lm",
+        ":is_nightly": "litert-lm-api-nightly",
+        "//conditions:default": "litert-lm-api",
     }),
     platform = select({
         "//build_config:linux_x86_64": "manylinux_2_35_x86_64",
@@ -262,5 +262,5 @@ py_wheel(
         ":is_nightly": VERSION + ".dev$(PYPI_NIGHTLY_DATE)",
         "//conditions:default": VERSION,
     }),
-    deps = [":litert_lm_pkg"],
+    deps = [":package"],
 )

python/litert_lm_cli/BUILD

@@ -73,17 +73,17 @@ config_setting(
 )
 
 py_package(
-    name = "litert_lm_cli_pkg",
+    name = "package",
     packages = ["python.litert_lm_cli"],
     deps = [":litert-lm"],
 )
 
 py_wheel(
-    name = "litert_lm_cli_wheel",
+    name = "wheel",
     abi = "none",
     distribution = select({
-        ":is_nightly": "litert-lm-cli-nightly",
-        "//conditions:default": "litert-lm-cli",
+        ":is_nightly": "litert-lm-nightly",
+        "//conditions:default": "litert-lm",
     }),
     platform = "any",
     python_tag = "py3",
@@ -110,10 +110,10 @@ py_wheel(
         ],
     },
     requires = select({
-        ":is_nightly": ["litert-lm-nightly"],
-        "//conditions:default": ["litert-lm"],
+        ":is_nightly": ["litert-lm-api-nightly"],
+        "//conditions:default": ["litert-lm-api"],
     }) + [
         "fire",
     ],
-    deps = [":litert_lm_cli_pkg"],
+    deps = [":package"],
 )