bug fix

PiperOrigin-RevId: 592794583 Change-Id: Ia6b6667d38da40eaacf034dbe147df70c1671587
google-deepmind · Dec 21, 2023 · 5cec6e5 · 5cec6e5
1 parent 28b8c0c
commit 5cec6e5
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/examples/cyberball/cyberball.ipynb b/examples/cyberball/cyberball.ipynb
@@ -250,14 +250,14 @@
         "      agent_name=agent_config.name,\n",
         "  )\n",
         "\n",
-        "  current_obs = components.observation.Observation(\n",
+        "  current_obs = agent_components.observation.Observation(\n",
         "            agent_name=agent_config.name,\n",
         "      clock_now=clock.now,\n",
         "      memory=mem,\n",
         "      timeframe=clock.get_step_size(),\n",
         "      component_name='current observations',\n",
         "  )\n",
-        "  summary_obs = components.observation.ObservationSummary(\n",
+        "  summary_obs = agent_components.observation.ObservationSummary(\n",
         "      agent_name=agent_config.name,\n",
         "      model=model,\n",
         "      clock_now=clock.now,\n",

diff --git a/examples/magic_beans_for_sale.ipynb b/examples/magic_beans_for_sale.ipynb
@@ -317,7 +317,7 @@
         "      memory=mem,\n",
         "      timeframe_delta_from=datetime.timedelta(hours=4),\n",
         "      timeframe_delta_until=datetime.timedelta(hours=1),\n",
-        "      components=[identity],\n",
+        "      components=[persona],\n",
         "      component_name='summary of observations',\n",
         "  )\n",
         "\n",