diff --git a/infra/k8s/argo/gpu-test-pipeline-template.yaml b/infra/k8s/argo/gpu-test-pipeline-template.yaml index 3af0be826..64498f028 100644 --- a/infra/k8s/argo/gpu-test-pipeline-template.yaml +++ b/infra/k8s/argo/gpu-test-pipeline-template.yaml @@ -263,7 +263,7 @@ spec: fi if [ "$SCOPE" = "integration" ] || [ "$SCOPE" = "all" ]; then run_tests "dqn-smoke" cargo test -p ml --features cuda --test smoke_test_real_data - run_tests "dqn-pipeline" cargo test -p ml --features cuda --test dqn_training_pipeline_test + run_tests "dqn-pipeline" cargo test -p ml --features cuda --test dqn_training_pipeline_test -- --test-threads=1 run_tests "dqn-smoke-train" cargo test -p ml --features cuda --test dqn_training_smoke_test run_tests "dqn-early-stop" cargo test -p ml --features cuda --test dqn_early_stopping_termination_test run_tests "dqn-collapse" cargo test -p ml --features cuda --test dqn_action_collapse_fix_test