Add special case in _get_tune_resources (#250)

* Adds special case in _get_tune_resources

Signed-off-by: Antoni Baum <antoni.baum@protonmail.com>

* Lint

Signed-off-by: Antoni Baum <antoni.baum@protonmail.com>

Signed-off-by: Antoni Baum <antoni.baum@protonmail.com>

Author: Yard1

xgboost_ray/tune.py

@@ -122,6 +122,11 @@ def _get_tune_resources(num_actors: int, cpus_per_actor: int,
         bundles = [head_bundle] + child_bundles
         placement_options = placement_options or {}
         placement_options.setdefault("strategy", "PACK")
+        # Special case, same as in
+        # ray.air.ScalingConfig.as_placement_group_factory
+        # TODO remove after Ray 2.3 is out
+        if placement_options.get("_max_cpu_fraction_per_node", None) is None:
+            placement_options.pop("_max_cpu_fraction_per_node", None)
         placement_group_factory = PlacementGroupFactory(
             bundles, **placement_options)