Remove parameters from discriminator optimizer update

Author: struan-robertson

examples/deep_convolutional_gan.ipynb

@@ -431,9 +431,7 @@
     "        key,\n",
     "    )\n",
     "\n",
-    "    updates, opt_state = discriminator_optimizer.update(\n",
-    "        grads, discriminator_opt_state, discriminator\n",
-    "    )\n",
+    "    updates, opt_state = discriminator_optimizer.update(grads, discriminator_opt_state)\n",
     "    discriminator = eqx.apply_updates(discriminator, updates)\n",
     "\n",
     "    return loss, discriminator, opt_state, generator_state, discriminator_state, key\n",