fix(infra): update GCP runner to v2.332.0 and resolve rebase import conflicts

Runner 2.317.0 does not support node24 required by actions/checkout@v6,
causing 'Set up job' failures on dynamically provisioned GCP instances.

Key changes:
- Bump RUNNER_VERSION from 2.317.0 to 2.332.0 in test-areal.yml and build-docker-image.yml
- Remove duplicate imports in test_rollout_controller.py from rebase with PR #996

Author: garrett4wade

.github/workflows/build-docker-image.yml

@@ -14,7 +14,7 @@ permissions:
 env:
   GCP_PROJECT_ID: ${{ secrets.GCP_PROJECT_ID }}
   VALIDATOR_LABELS: gcp-docker-validator
-  RUNNER_VERSION: '2.317.0'
+  RUNNER_VERSION: '2.332.0'
   IMAGE_NAME: ghcr.io/inclusionai/areal-runtime
   IMAGE_TAG: test
 

.github/workflows/test-areal.yml

@@ -39,7 +39,7 @@ concurrency:
 
 env:
   GCP_PROJECT_ID: ${{ secrets.GCP_PROJECT_ID }}
-  RUNNER_VERSION: '2.317.0'
+  RUNNER_VERSION: '2.332.0'
   GCP_OS_IMAGE: areal-cicd-test-202602030
 
 jobs:

tests/grpo/config_fsdp.yaml tests/grpo/config.yamltests/grpo/config_fsdp.yaml renamed to tests/grpo/config.yaml

@@ -1,5 +1,9 @@
+# Unified GRPO test config.
+# Backend-specific values (allocation_mode, optimizer) are overridden via CLI.
 experiment_name: tests-grpo
 trial_name: trial
+scheduler:
+  type: local
 cluster:
   n_nodes: 1
   n_gpus_per_node: 2
@@ -33,6 +37,10 @@ recover:
 sglang:
   model_path: ${actor.path}
   random_seed: ${seed}
+vllm:
+  model: ${actor.path}
+  seed: ${seed}
+  skip_tokenizer_init: false
 gconfig:
   n_samples: 4
   max_new_tokens: 1024
@@ -42,6 +50,7 @@ rollout:
   trial_name: ${trial_name}
   consumer_batch_size: ${train_dataset.batch_size}
   scheduling_spec: ${actor.scheduling_spec}
+  max_head_offpolicyness: 2
 
 actor:
   experiment_name: ${experiment_name}