Release v0.1

GPT4_game.ipynb

@@ -7,7 +7,7 @@
         "colab_type": "text"
       },
       "source": [
-        "<a href=\"https://colab.research.google.com/github/srush/GPTWorld-Challenge/blob/main/GPT4_game.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>"
+        "<a href=\"https://colab.research.google.com/github/srush/GPTWorld/blob/main/GPT4_game.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>"
       ]
     },
     {
@@ -38,12 +38,13 @@
     },
     {
       "cell_type": "code",
-      "execution_count": 44,
+      "execution_count": null,
       "metadata": {
         "id": "gKzCE5A7aGWY"
       },
       "outputs": [],
       "source": [
+        "#@title Install (collapse me)\n",
         "%%capture\n",
         "!sudo apt-get install libcairo2-dev\n",
         "!pip install -U git+https://github.com/chalk-diagrams/chalk openai pycairo tiktoken"
@@ -60,13 +61,14 @@
     },
     {
       "cell_type": "code",
-      "execution_count": 2,
+      "execution_count": null,
       "metadata": {
-        "id": "RWUExPo7Y8-O"
+        "id": "RWUExPo7Y8-O",
+        "cellView": "form"
       },
       "outputs": [],
       "source": [
-        "\n",
+        "#@title Import (collapse me)\n",
         "from dataclasses import dataclass\n",
         "from chalk import *\n",
         "from colour import Color\n",
@@ -133,12 +135,15 @@
     },
     {
       "cell_type": "code",
-      "execution_count": 54,
+      "execution_count": null,
       "metadata": {
-        "id": "Fv3eTRKiV2ZB"
+        "id": "Fv3eTRKiV2ZB",
+        "cellView": "form"
       },
       "outputs": [],
       "source": [
+        "#@title Game Code\n",
+        "\n",
         "# Possible Actions\n",
         "class Actions(Enum):\n",
         "    UPRIGHT = \"UR\"\n",
@@ -256,6 +261,7 @@
     {
       "cell_type": "code",
       "source": [
+        "#@title Drawing code\n",
         "def draw_board(grid, num=0):\n",
         "    hex = regular_polygon(6, 1).rotate_by(1/12).line_width(0.5).fill_color(Color(\"white\"))\n",
         "    w = hex.get_envelope().width\n",
@@ -291,7 +297,7 @@
         "id": "Ic7WgOTi8uF1",
         "outputId": "4dc07cb9-9e5f-4d28-d4ea-470ad4b13141"
       },
-      "execution_count": 76,
+      "execution_count": null,
       "outputs": [
         {
           "output_type": "execute_result",
@@ -367,7 +373,7 @@
         "id": "nqgPKLu0AMhU",
         "outputId": "19e4c6d0-b792-4a34-f4c4-81902974c346"
       },
-      "execution_count": 78,
+      "execution_count": null,
       "outputs": [
         {
           "output_type": "display_data",
@@ -502,13 +508,13 @@
     },
     {
       "cell_type": "code",
-      "execution_count": 102,
+      "execution_count": null,
       "metadata": {
         "id": "jFf7TCOJaVHX"
       },
       "outputs": [],
       "source": [
-        "\n",
+        "#@title Make the Prompt\n",
         "def make_fun(board, actions):\n",
         "    \"This function generates python code for few-shot examples\"\n",
         "    out = tab + \"p = \" + str(board.player_pos)\n",
@@ -607,7 +613,7 @@
         "id": "cOneYFok_OMe",
         "outputId": "97080186-7322-4ba9-b500-095fb39071aa"
       },
-      "execution_count": 103,
+      "execution_count": null,
       "outputs": [
         {
           "output_type": "stream",
@@ -669,7 +675,7 @@
     },
     {
       "cell_type": "code",
-      "execution_count": 105,
+      "execution_count": null,
       "metadata": {
         "id": "LONWUsBLjOHo",
         "colab": {
@@ -967,7 +973,7 @@
   "metadata": {
     "colab": {
       "provenance": [],
-      "authorship_tag": "ABX9TyMCVv6YyI0lCdACYaFERyNc",
+      "authorship_tag": "ABX9TyNFpIRGCX0LXC26SCWsfp30",
       "include_colab_link": true
     },
     "kernelspec": {