wip

Signed-off-by: sven1977 <[email protected]>

rllib/utils/test_utils.py

@@ -1156,9 +1156,6 @@ def run_rllib_example_script_experiment(
                     config.learners(num_gpus_per_learner=1)
                 else:
                     config.learners(num_gpus_per_learner=0)
-            if args.num_cpus_per_learner is not None:
-                config.learners(num_cpus_per_learner=args.num_cpus_per_learner)
-
             # User hard-requires n GPUs, but they are not available -> Error.
             elif num_gpus_available < num_gpus_requested:
                 raise ValueError(
@@ -1173,6 +1170,10 @@ def run_rllib_example_script_experiment(
             else:
                 config.learners(num_gpus_per_learner=args.num_gpus_per_learner)
 
+            # Set CPUs per Learner.
+            if args.num_cpus_per_learner is not None:
+                config.learners(num_cpus_per_learner=args.num_cpus_per_learner)
+
         # Old stack (override only if arg was provided by user).
         elif args.num_gpus is not None:
             config.resources(num_gpus=args.num_gpus)