Skip to content

Commit

Permalink
fix: changes after review
Browse files Browse the repository at this point in the history
  • Loading branch information
AntoineRR committed Feb 7, 2023
1 parent 04c90b8 commit b863136
Show file tree
Hide file tree
Showing 2 changed files with 8 additions and 21 deletions.
12 changes: 0 additions & 12 deletions robyn/argument_parser.py
Original file line number Diff line number Diff line change
Expand Up @@ -49,15 +49,3 @@ def __init__(self) -> None:
self.log_level = "INFO"
else:
self.log_level = args.log_level

def as_str(self):
return [
"--processes",
f"{self.processes}",
"--workers",
f"{self.workers}",
"--dev",
f"{self.dev}",
"--log-level",
f"{self.log_level}",
]
17 changes: 8 additions & 9 deletions robyn/logger.py
Original file line number Diff line number Diff line change
Expand Up @@ -11,13 +11,12 @@ class Colors(Enum):
RED = "\033[91m"


HEADER = "\033[95m"
ENDC = "\033[0m"
BOLD = "\033[1m"
UNDERLINE = "\033[4m"


class Logger:
HEADER = "\033[95m"
ENDC = "\033[0m"
BOLD = "\033[1m"
UNDERLINE = "\033[4m"

def __init__(self):
self.logger = logging.getLogger(__name__)

Expand All @@ -30,11 +29,11 @@ def _format_msg(
):
result = msg
if color is not None:
result = f"{color.value}{result}{ENDC}"
result = f"{color.value}{result}{Logger.ENDC}"
if bold:
result = f"{BOLD}{result}"
result = f"{Logger.BOLD}{result}"
if underline:
result = f"{UNDERLINE}{result}"
result = f"{Logger.UNDERLINE}{result}"
return result

def error(
Expand Down

0 comments on commit b863136

Please sign in to comment.