[air/hotfix] Fix test_tune (ray-project#26721)

Broken by ray-project#26493 Signed-off-by: Kai Fricke <[email protected]> Signed-off-by: Stefan van der Kleij <[email protected]>
Stefan-1313 · Aug 18, 2022 · 5619cb0 · 5619cb0
1 parent 2419667
commit 5619cb0
Showing 1 changed file with 5 additions and 6 deletions.
diff --git a/python/ray/train/tests/test_tune.py b/python/ray/train/tests/test_tune.py
@@ -335,13 +335,12 @@ def train_func():
     trainer = TorchTrainer(
         train_func,
         torch_config=TorchConfig(backend="gloo"),
-        scaling_config={
-            "num_workers": 2,
-            "use_gpu": True,
-            "resources_per_worker": {"GPU": num_gpus_per_worker},
-        },
+        scaling_config=ScalingConfig(
+            num_workers=2,
+            use_gpu=True,
+            resources_per_worker={"GPU": num_gpus_per_worker},
+        ),
     )
-
     tuner = Tuner(
         trainer,
         param_space={