From c9ace03f2d1ca0b36215dd945fc91b2933c3b0c7 Mon Sep 17 00:00:00 2001
From: xlliu-scitix <xlliu@scitix.ai>
Date: Sun, 21 Dec 2025 08:12:47 +0000
Subject: [PATCH] update pre-check.yml

---
 .github/workflows/pre-check.yml        | 2 +-
 docker/Dockerfile.cuda13.x.ubuntu22.04 | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/.github/workflows/pre-check.yml b/.github/workflows/pre-check.yml
index 0b8d414..f8f75d0 100644
--- a/.github/workflows/pre-check.yml
+++ b/.github/workflows/pre-check.yml
@@ -21,7 +21,7 @@ jobs:
           docker buildx build \
             -f docker/Dockerfile.cuda13.x.ubuntu22.04 \
             --platform linux/amd64 \
-            --target package \
+            --target build \
             --output type=local,dest=dist \
             .
       - name: List artifacts
diff --git a/docker/Dockerfile.cuda13.x.ubuntu22.04 b/docker/Dockerfile.cuda13.x.ubuntu22.04
index 8e6a453..1ca8a1c 100644
--- a/docker/Dockerfile.cuda13.x.ubuntu22.04
+++ b/docker/Dockerfile.cuda13.x.ubuntu22.04
@@ -24,7 +24,7 @@ ARG BUILD_DATE=20251221
 ###########################
 # Build Stage
 ###########################
-FROM nvidia/cuda:${CUDA_VERSION}-devel-ubuntu${UBUNTU_VERSION}
+FROM nvidia/cuda:${CUDA_VERSION}-devel-ubuntu${UBUNTU_VERSION} AS build
 
 # Re-declare build args for this stage (values are inherited)
 ARG UBUNTU_VERSION