fix: Gumbel-Softmax notebook optimizers update

ydataai · Apr 19, 2022 · 6fdb4b1 · 6fdb4b1
1 parent e2f4159
commit 6fdb4b1
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/examples/regular/gumbel_softmax_example.ipynb b/examples/regular/gumbel_softmax_example.ipynb
@@ -147,7 +147,8 @@
     "    This makes it equivalent to the previous implementation.\n",
     "    The source code will help you understanding how it works\"\"\"\n",
     "    def define_gan(self, activation_info = None):\n",
-    "        super().define_gan(activation_info=None)"
+    "        optimizers = super().define_gan(activation_info=None)\n",
+    "        return optimizers"
    ]
   },
   {
@@ -2166,7 +2167,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.8.11"
+   "version": "3.8.13"
   },
   "orig_nbformat": 4
  },