[CICD] Expand unit test coverage on CUDA

.github/configs/cuda.yml

@@ -3,7 +3,7 @@
 # Test configurations are defined in tests/test_utils/config/platforms/cuda.yaml
 
 hardware_name: cuda
-display_name: "CUDA Tests"
+display_name: 'CUDA Tests'
 
 # Docker image for this hardware
 ci_image: harbor.baai.ac.cn/flagos/flagscale:cuda12.8.1-cudnn9.7.1-python3.12-torch2.7.0-time2507111538
@@ -52,14 +52,14 @@ container_options: >-
 #   2. Ensure uv is installed in the Docker image
 #   3. Set env_path to the virtual environment path (e.g., "/opt/venv")
 #
-pkg_mgr: "conda"  # Current: conda for CI/CD compatibility
+pkg_mgr: 'conda' # Current: conda for CI/CD compatibility
 
 # Environment path (venv path for uv, conda installation path for conda)
-env_path: "/root/miniconda3"
+env_path: '/root/miniconda3'
 
 # Conda environment name (for conda only)
 env_names:
-  train: "flagscale-train"
-  hetero_train: "flagscale-train"
-  inference: "flagscale-inference"
-  rl: "flagscale-rl"
+  train: 'flagscale-train'
+  hetero_train: 'flagscale-train'
+  inference: 'flagscale-inference'
+  rl: 'flagscale-rl'

.github/configs/metax.yml

@@ -41,14 +41,14 @@ container_options: >-
 # =============================================================================
 # Package Manager Configuration
 # =============================================================================
-pkg_mgr: "conda"
+pkg_mgr: 'conda'
 
 # Environment path (conda installation path)
-env_path: "/opt/conda"
+env_path: '/opt/conda'
 
 # Conda environment names
 env_names:
-  train: "base"
-  hetero_train: ""
-  inference: ""
-  rl: ""
+  train: 'base'
+  hetero_train: ''
+  inference: ''
+  rl: ''

.github/workflows/build_image_cuda.yml

@@ -49,14 +49,14 @@ on:
         required: false
         type: string
         description: Override runs_on. Falls back to platform config if not set.
-        default: '["self-hosted", "Linux", "X64", "nvidia-0", "gpus-8"]'
+        default: '["flagscale-nvidia-a100-gpu2-32c-128g"]'
       container_volumes:
         required: false
         type: string
         description: Override container_volumes. Falls back to platform config if not set.
         default: >-
-          ["/home/flagscale_cicd/docker/docker_build/docker_data:/home/gitlab-runner/data",
-          "/home/flagscale_cicd/docker/docker_build/docker_tokenizers:/home/gitlab-runner/tokenizers"]
+          ["/mnt/airs-business/cicd/baai_datasets:/home/gitlab-runner/data",
+          "/mnt/airs-business/cicd/baai_tokenizers:/home/gitlab-runner/tokenizers"]
 
   # Trigger on PRs that modify docker-related files (build + test only, no push)
   pull_request:
@@ -77,7 +77,7 @@ concurrency:
 
 env:
   # Local registry used by CI runners (see .github/configs/cuda.yml)
-  REGISTRY: localhost:5000
+  REGISTRY: harbor.baai.ac.cn/flagscale
   # Default build versions (keep in sync with docker/build.sh)
   CUDA_VERSION: '12.8.1'
   UBUNTU_VERSION: '22.04'
@@ -151,7 +151,7 @@ jobs:
   build:
     name: Build ${{ matrix.task }}
     needs: prepare
-    runs-on: [self-hosted, Linux, X64, nvidia-0, gpus-8]
+    runs-on: ${{ fromJson(inputs.runs_on || '["flagscale-nvidia-a100-gpu2-32c-128g"]') }}
     env:
       HTTP_PROXY: ${{ vars.HTTP_PROXY }}
       HTTPS_PROXY: ${{ vars.HTTPS_PROXY }}
@@ -316,7 +316,7 @@ jobs:
   load_images:
     name: Load and push images
     needs: ['build', 'summary']
-    runs-on: [self-hosted, Linux, X64, nvidia-0, gpus-8]
+    runs-on: ${{ fromJson(inputs.runs_on || '["flagscale-nvidia-a100-gpu2-32c-128g"]') }}
     steps:
       - name: Load train image from tar and push
         run: |
@@ -354,9 +354,8 @@ jobs:
       ci_train_image: ${{ needs.build.outputs.train_tag }}
       ci_inference_image: ${{ needs.build.outputs.inference_tag }}
       runs_on: >-
-        ${{ inputs.runs_on ||
-        '["self-hosted", "Linux", "X64", "nvidia-0", "gpus-8"]' }}
+        ${{ inputs.runs_on || '["flagscale-nvidia-a100-gpu2-32c-128g"]' }}
       container_volumes: >-
         ${{ inputs.container_volumes ||
-        '["/home/flagscale_cicd/docker/docker_build/docker_data:/home/gitlab-runner/data",
-        "/home/flagscale_cicd/docker/docker_build/docker_tokenizers:/home/gitlab-runner/tokenizers"]' }}
+        '["/mnt/airs-business/cicd/docker_data:/home/gitlab-runner/data",
+        "/mnt/airs-business/cicd/docker_tokenizers:/home/gitlab-runner/tokenizers"]' }}

tests/test_utils/runners/run_unit_tests.sh

@@ -94,7 +94,17 @@ run_unit_tests_for_device() {
         cat > "$COVERAGERC" <<EOF
 [run]
 parallel = true
-source = $PROJECT_ROOT
+source =
+    $PROJECT_ROOT/flagscale
+omit =
+    */tools/*
+    */tests/*
+    */examples/*
+    */docs/*
+    */docker/*
+    */setup.py
+    */run.py
+    */__init__.py
 data_file = $COVERAGE_DIR/.coverage
 EOF
     fi