Skip to content

GitLab

Unverified Commit 67db63a5 authored by Alex Titterton's avatar Alex Titterton Committed by GitHub
Browse files

wandb logger fix, added pre-commit (#2484)

parent 6b628d9a
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 3 deletions
lm_eval/loggers/wandb_logger.py
View file @ 67db63a5
...@@ -15,10 +15,9 @@ logger = logging.getLogger(__name__) ...@@ -15,10 +15,9 @@ logger = logging.getLogger(__name__)
def get_wandb_printer() -> Literal["Printer"]: def get_wandb_printer() -> Literal["Printer"]:
"""Returns a wandb printer instance for pretty stdout.""" """Returns a wandb printer instance for pretty stdout."""
from wandb.sdk.lib.printer import get_printer from wandb.sdk.lib.printer import new_printer
from wandb.sdk.wandb_settings import Settings
printer = get_printer(Settings()._jupyter) printer = new_printer()
return printer return printer
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment