remove device in Config (deprecation)

leopard-ai · Oct 10, 2023 · c6105e2 · c6105e2
1 parent c15d46b
commit c6105e2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/examples/logistic_regression_hpo/test.py b/examples/logistic_regression_hpo/test.py
@@ -109,8 +109,8 @@ def on_inner_loop_start(self):
 child_config = Config(type="cg", cg_iterations=3, cg_alpha=0.1, unroll_steps=100)
 engine_config = EngineConfig(train_iters=5000)
 
-parent = Parent(name="outer", config=parent_config, device=device)
-child = Child(name="inner", config=child_config, device=device)
+parent = Parent(name="outer", config=parent_config)
+child = Child(name="inner", config=child_config)
 
 problems = [parent, child]
 u2l = {parent: [child]}