Skip to content

Commit

Permalink
make style
Browse files Browse the repository at this point in the history
  • Loading branch information
NathanHB committed Jul 8, 2024
1 parent ce28dca commit 48c3857
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 2 deletions.
2 changes: 1 addition & 1 deletion src/lighteval/logging/evaluation_tracker.py
Original file line number Diff line number Diff line change
Expand Up @@ -495,7 +495,7 @@ def push_to_tensorboard( # noqa: C901
if not is_tensorboardX_available:
hlog_warn(NO_TENSORBOARDX_WARN_MSG)
return

if not is_nanotron_available():
hlog_warn("You cannot push results to tensorboard without having nanotron installed. Skipping")
return
Expand Down
2 changes: 1 addition & 1 deletion src/lighteval/utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -191,7 +191,6 @@ def is_peft_available() -> bool:
NO_PEFT_ERROR_MSG = "You are trying to use adapter weights models, for which you need `peft`, which is not available in your environment. Please install it using pip."



def is_tensorboardX_available() -> bool:
return importlib.util.find_spec("tensorboardX") is not None

Expand All @@ -200,6 +199,7 @@ def is_tensorboardX_available() -> bool:
"You are trying to log using tensorboardX, which is not installed. Please install it using pip. Skipping."
)


def is_openai_available() -> bool:
return importlib.util.find_spec("openai") is not None

Expand Down

0 comments on commit 48c3857

Please sign in to comment.