Fix remaining usages

icaros-usc · Nov 2, 2023 · 9e726d1 · 9e726d1
1 parent bc1ec94
commit 9e726d1
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 6 deletions.
diff --git a/ribs/archives/_archive_data_frame.py b/ribs/archives/_archive_data_frame.py
@@ -84,7 +84,7 @@ def iterelites(self):
         """Iterator that outputs every elite in the ArchiveDataFrame.
 
         Data which is unavailable will be turned into None. For example, if
-        there are no solution columns, then ``elite.solution`` will be None.
+        there are no solution columns, then ``elite["solution"]`` will be None.
         """
         solution_batch = self.solution_batch()
         objective_batch = self.objective_batch()

diff --git a/tutorials/arm_repertoire.ipynb b/tutorials/arm_repertoire.ipynb
@@ -451,8 +451,8 @@
    "source": [
     "elite = archive.retrieve_single([0, 0])\n",
     "_, ax = plt.subplots()\n",
-    "if elite.solution is not None:  # This is None if there is no solution for [0,0].\n",
-    "    visualize(elite.solution, link_lengths, elite.objective, ax)"
+    "if elite[\"solution\"] is not None:  # This is None if there is no solution for [0,0].\n",
+    "    visualize(elite[\"solution\"], link_lengths, elite[\"objective\"], ax)"
    ]
   },
   {

diff --git a/tutorials/fooling_mnist.ipynb b/tutorials/fooling_mnist.ipynb
@@ -303,13 +303,13 @@
     "\n",
     "# Display images.\n",
     "for elite in archive:\n",
-    "    digit = elite.index\n",
+    "    digit = elite[\"index\"]\n",
     "    found.add(digit)\n",
     "\n",
     "    # No need to normalize image because we want to see the original.\n",
-    "    ax[digit].imshow(elite.solution.reshape(28, 28), cmap=\"Greys\")\n",
+    "    ax[digit].imshow(elite[\"solution\"].reshape(28, 28), cmap=\"Greys\")\n",
     "    ax[digit].set_axis_off()\n",
-    "    ax[digit].set_title(f\"{digit} | Score: {elite.objective:.3f}\", pad=8)\n",
+    "    ax[digit].set_title(f\"{digit} | Score: {elite['objective']:.3f}\", pad=8)\n",
     "\n",
     "# Mark digits that we did not generate images for.\n",
     "for digit in range(10):\n",