Skip to content

Commit

Permalink
linter formatting
Browse files Browse the repository at this point in the history
  • Loading branch information
brifordwylie committed Jan 1, 2024

Verified

This commit was signed with the committer’s verified signature.
sxzz Kevin Deng 三咲智子
1 parent d6912fb commit f78c2e9
Showing 2 changed files with 3 additions and 6 deletions.
6 changes: 2 additions & 4 deletions src/sageworks/repl/sageworks_shell.py
Original file line number Diff line number Diff line change
@@ -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"""
3 changes: 1 addition & 2 deletions src/sageworks/utils/repl_utils.py
Original file line number Diff line number Diff line change
@@ -28,6 +28,5 @@ def cprint(text, color):


if __name__ == "__main__":

for color in colors.keys():
cprint(f"Hello world! ({color})", color)
cprint(f"Hello world! ({color})", color)

0 comments on commit f78c2e9

Please sign in to comment.