Save wandb username and project name into env vars

Author: minhna4

.env

@@ -0,0 +1,2 @@
+WANDB_PROJECT_NAME="tennis-maddpg"
+WANDB_USER_NAME="minhna1112"

Tennis.ipynb

@@ -300,7 +300,7 @@
     "\n",
     "from utils import Config, ReplayBuffer\n",
     "from agents import MADDPG \n",
-    "\n",
+    "from load_env import env_vars\n",
     "env = UnityEnvironment(file_name=\"./Tennis_Linux_NoVis/Tennis.x86_64\")\n",
     "# get the default brain\n",
     "brain_name = env.brain_names[0]\n",
@@ -342,7 +342,8 @@
     "SOLVED_SCORE = 0.5\n",
     "\n",
     "# Initialize wandb run\n",
-    "wandb.init(project=\"tennis-maddpg\", entity=\"minhna1112\")  # Replace 'your_username' with your wandb username\n",
+    "wandb.init(project=env_vars.WANDB_PROJECT_NAME, \n",
+    "           entity=env_vars.WANDB_USER_NAME)  # Replace 'your_username' with your wandb username\n",
     "\n",
     "model_dir= os.getcwd()+\"/checkpoints\"\n",
     "os.makedirs(model_dir, exist_ok=True)\n",
@@ -422,7 +423,17 @@
    "execution_count": null,
    "metadata": {},
    "outputs": [],
-   "source": []
+   "source": [
+    "fig = plt.figure(figsize=(10,4))\n",
+    "ax = fig.add_subplot(111)\n",
+    "plt.plot(np.arange(1, len(scores_all)+1), scores_all, label='Scores')\n",
+    "plt.plot(np.arange(1, len(moving_average)+1), moving_average, c='r', label='Average')\n",
+    "plt.ylabel('Score')\n",
+    "plt.xlabel('Episode #')\n",
+    "ax.legend(fontsize='large', loc='upper left')\n",
+    "fig.savefig('result.png')\n",
+    "plt.show()"
+   ]
   }
  ],
  "metadata": {

load_env.py

@@ -0,0 +1,9 @@
+from dotenv import load_dotenv
+import os
+
+load_dotenv("./env")
+class ENV:
+    WANDB_PROJECT_NAME = os.getenv('WANDB_PROJECT_NAME')
+    WANDB_USER_NAME = os.getenv('WANDB_USER_NAME')
+
+env_vars = ENV()