Merge pull request #784 from NVIDIA/am/stable-tests

Make tests more stable on systems with slurm binaries

Author: amaslenn

tests/test_acceptance.py

@@ -525,6 +525,7 @@ def test_req(request, slurm_system: SlurmSystem, partial_tr: partial[TestRun]) -
 def test_sbatch_generation(slurm_system: SlurmSystem, test_req: tuple[TestRun, str]):
     slurm_system.output_path.mkdir(parents=True, exist_ok=True)
     slurm_system.container_mount_home = True
+    slurm_system.supports_gpu_directives_cache = True
 
     tr = test_req[0]
 

tests/test_single_sbatch_runner.py

@@ -1,5 +1,5 @@
 # SPDX-FileCopyrightText: NVIDIA CORPORATION & AFFILIATES
-# Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# Copyright (c) 2025-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 # SPDX-License-Identifier: Apache-2.0
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
@@ -100,6 +100,7 @@ def test_sbatch_system_fields(sleep_tr: TestRun, slurm_system: SlurmSystem) -> N
 
     runner.system.account = "test_account"
     runner.system.distribution = "test_distribution"
+    runner.system.supports_gpu_directives_cache = True
     runner.system.gpus_per_node = 2
     runner.system.ntasks_per_node = 4
     runner.system.extra_sbatch_args = ["--test-arg1", "--test-arg2"]