update testing and requirements

Author: gulsumgudukbay

.github/workflows/run_tests_coordinator.yml

@@ -125,8 +125,8 @@ jobs:
             "gpu-integration": "not cpu_only and not tpu_only and integration_test and not post_training",
             "cpu-unit": "cpu_only and not post_training",
             "cpu-post-training-unit": "cpu_only and post_training",
-            "rocm-unit": "not cpu_only and not tpu_only and not decoupled",
-            "rocm-decoupled": "not cpu_only and not tpu_only and decoupled"
+            "rocm-unit": "not cpu_only and not tpu_only and not post_training and not decoupled",
+            "rocm-decoupled": "not cpu_only and not tpu_only and not post_training and decoupled"
           }')[inputs.flavor] }}
 
       pytest_addopts: >-
@@ -153,8 +153,8 @@ jobs:
             "gpu-integration": "--ignore=tests/post_training",
             "cpu-unit": "--ignore=tests/post_training",
             "cpu-post-training-unit": "",
-            "rocm-unit": "",
-            "rocm-decoupled": ""
+            "rocm-unit": "--ignore=tests/post_training",
+            "rocm-decoupled": "--ignore=tests/post_training"
           }')[inputs.flavor] }}
         ${{ inputs.additional_pytest_args }}
 
@@ -184,7 +184,7 @@ jobs:
           }')[inputs.flavor] || '--privileged' }}
 
       # ROCm-specific parameters
-      decoupled_mode: ${{ contains(inputs.flavor, 'rocm') }}
+      decoupled_mode: ${{ contains(inputs.flavor, 'decoupled') }}
       requirements_file: >-
         ${{ fromJSON('{
             "rocm-unit": "src/dependencies/requirements/requirements_rocm_jax_0_9_1.txt",

src/dependencies/requirements/requirements_decoupled_rocm_jax_0_9_1.txt

@@ -11,11 +11,6 @@ jax==0.9.1
 # ROCm JAX wheels
 jax-rocm7-plugin==0.9.1.*
 
-# Post-training deps (needed at pytest collection time for tests/post_training/unit/*).
-# vLLM ROCm wheel pinned to match the rocm7.1.2 / gfx950 stack. Installed via direct
-# URL (instead of --extra-index-url) so AMD's flat index isn't probed for unrelated
-# packages (e.g. setuptools during sdist builds), which would 403 and break resolution.
-vllm @ https://rocm.frameworks.amd.com/whl/gfx950-dcgpu/vllm-0.16.1.dev10+g11515110f.d20260324.rocm712-cp312-cp312-linux_x86_64.whl
 math-verify
 
 jaxtyping

src/dependencies/requirements/requirements_rocm_jax_0_9_1.txt

@@ -11,11 +11,6 @@ jax==0.9.1
 # ROCm JAX wheels
 jax-rocm7-plugin==0.9.1.*
 
-# Post-training deps (needed at pytest collection time for tests/post_training/unit/*).
-# vLLM ROCm wheel pinned to match the rocm7.1.2 / gfx950 stack. Installed via direct
-# URL (instead of --extra-index-url) so AMD's flat index isn't probed for unrelated
-# packages (e.g. setuptools during sdist builds), which would 403 and break resolution.
-vllm @ https://rocm.frameworks.amd.com/whl/gfx950-dcgpu/vllm-0.16.1.dev10+g11515110f.d20260324.rocm712-cp312-cp312-linux_x86_64.whl
 math-verify