tweaks (TODO: fix in follow up PR)

GFNOrg · Nov 29, 2023 · f12cbec · f12cbec
1 parent 25b7527
commit f12cbec
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tutorials/notebooks/intro_gfn_smiley.ipynb b/tutorials/notebooks/intro_gfn_smiley.ipynb
@@ -1728,7 +1728,7 @@
         "def train(gflownet, optimizer, env, batch_size = 128, n_episodes = 25_000):\n",
         "    \"\"\"Training loop, keeping track of terminal states over training.\"\"\"\n",
         "    # This stores example terminating states.\n",
-        "    visited_terminating_states = env.States.from_batch_shape((0,))\n",
+        "    visited_terminating_states = env.states_from_batch_shape((0,))\n",
         "    states_visited = 0\n",
         "    losses = []\n",
         "\n",