Update opt.minimize args

PennyLaneAI · Aug 30, 2023 · 6a8a897 · 6a8a897
1 parent 71d6d3a
commit 6a8a897
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/demonstrations/tutorial_QGAN.py b/demonstrations/tutorial_QGAN.py
@@ -213,7 +213,7 @@ def gen_cost(gen_weights):
 cost = lambda: disc_cost(disc_weights)
 
 for step in range(50):
-    opt.minimize(cost, disc_weights)
+    opt.minimize(cost, [disc_weights])
     if step % 5 == 0:
         cost_val = cost().numpy()
         print("Step {}: cost = {}".format(step, cost_val))
@@ -243,7 +243,7 @@ def gen_cost(gen_weights):
 cost = lambda: gen_cost(gen_weights)
 
 for step in range(50):
-    opt.minimize(cost, gen_weights)
+    opt.minimize(cost, [gen_weights])
     if step % 5 == 0:
         cost_val = cost().numpy()
         print("Step {}: cost = {}".format(step, cost_val))