From 7da48846a5cb843eaaf41d9e42bbe14940afd5d5 Mon Sep 17 00:00:00 2001 From: Andres Caicedo Date: Thu, 27 Apr 2023 09:56:06 +0200 Subject: [PATCH 1/2] Optimize cprint function using dictionary Replace multiple if-elif statements in cprint function with a dictionary to store color codes for each color, resulting in a more concise and maintainable code. Also added a check for invalid colors. --- notebooks/stablelm-alpha.ipynb | 20 ++++++++++++-------- 1 file changed, 12 insertions(+), 8 deletions(-) diff --git a/notebooks/stablelm-alpha.ipynb b/notebooks/stablelm-alpha.ipynb index 457adc3..7a3ed13 100644 --- a/notebooks/stablelm-alpha.ipynb +++ b/notebooks/stablelm-alpha.ipynb @@ -54,14 +54,18 @@ "\n", "from IPython.display import Markdown, display\n", "def hr(): display(Markdown('---'))\n", - "def cprint(msg: str, color: str = \"blue\", **kwargs) -> str:\n", - " if color == \"blue\": print(\"\\033[34m\" + msg + \"\\033[0m\", **kwargs)\n", - " elif color == \"red\": print(\"\\033[31m\" + msg + \"\\033[0m\", **kwargs)\n", - " elif color == \"green\": print(\"\\033[32m\" + msg + \"\\033[0m\", **kwargs)\n", - " elif color == \"yellow\": print(\"\\033[33m\" + msg + \"\\033[0m\", **kwargs)\n", - " elif color == \"purple\": print(\"\\033[35m\" + msg + \"\\033[0m\", **kwargs)\n", - " elif color == \"cyan\": print(\"\\033[36m\" + msg + \"\\033[0m\", **kwargs)\n", - " else: raise ValueError(f\"Invalid info color: `{color}`\")" + "def cprint(msg: str, color: str = \"blue\", **kwargs) -> None:\n", + " color_codes = {\n", + " \"blue\": \"\\033[34m\",\n", + " \"red\": \"\\033[31m\",\n", + " \"green\": \"\\033[32m\",\n", + " \"yellow\": \"\\033[33m\",\n", + " \"purple\": \"\\033[35m\",\n", + " \"cyan\": \"\\033[36m\",\n", + " }\n", + " \n", + " if color not in color_codes:\n", + " raise ValueError(f\"Invalid info color: `{color}`\")" ] }, { From 8c00a3c2baf95f447ae26865e5b886d8ac1e1cc5 Mon Sep 17 00:00:00 2001 From: Andres Caicedo Date: Thu, 27 Apr 2023 09:59:15 +0200 Subject: [PATCH 2/2] Update stablelm-alpha.ipynb Print --- notebooks/stablelm-alpha.ipynb | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/notebooks/stablelm-alpha.ipynb b/notebooks/stablelm-alpha.ipynb index 7a3ed13..82494bb 100644 --- a/notebooks/stablelm-alpha.ipynb +++ b/notebooks/stablelm-alpha.ipynb @@ -65,7 +65,9 @@ " }\n", " \n", " if color not in color_codes:\n", - " raise ValueError(f\"Invalid info color: `{color}`\")" + " raise ValueError(f\"Invalid info color: `{color}`\")\n", + " \n", + " print(color_codes[color] + msg + \"\\033[0m\", **kwargs)" ] }, {