From f78c2e97d0e18f71a9d50babdbd9d24fc55a07b7 Mon Sep 17 00:00:00 2001 From: Brian Wylie Date: Mon, 1 Jan 2024 12:08:49 -0700 Subject: [PATCH] linter formatting --- src/sageworks/repl/sageworks_shell.py | 6 ++---- src/sageworks/utils/repl_utils.py | 3 +-- 2 files changed, 3 insertions(+), 6 deletions(-) diff --git a/src/sageworks/repl/sageworks_shell.py b/src/sageworks/repl/sageworks_shell.py index 2f67fcf0c..cc82d6007 100644 --- a/src/sageworks/repl/sageworks_shell.py +++ b/src/sageworks/repl/sageworks_shell.py @@ -12,6 +12,7 @@ from sageworks.utils.repl_utils import cprint import sageworks # noqa: F401 from sageworks.utils.sageworks_logging import IMPORTANT_LEVEL_NUM + logging.getLogger("sageworks").setLevel(logging.WARNING) @@ -55,10 +56,7 @@ class SageWorksPrompt(Prompts): def in_prompt_tokens(self, cli=None): aws_profile = os.getenv("AWS_PROFILE", "default") # return [(Token.Prompt, "🍺 "), (Token.Prompt, f"SageWorks({aws_profile})> ")] - return [ - (Token.SageWorks, "SageWorks"), - (Token.AWSProfile, f"({aws_profile})> ") - ] + return [(Token.SageWorks, "SageWorks"), (Token.AWSProfile, f"({aws_profile})> ")] def start(self): """Start the SageWorks IPython shell""" diff --git a/src/sageworks/utils/repl_utils.py b/src/sageworks/utils/repl_utils.py index 054385a37..95309d305 100644 --- a/src/sageworks/utils/repl_utils.py +++ b/src/sageworks/utils/repl_utils.py @@ -28,6 +28,5 @@ def cprint(text, color): if __name__ == "__main__": - for color in colors.keys(): - cprint(f"Hello world! ({color})", color) \ No newline at end of file + cprint(f"Hello world! ({color})", color)