Skip to content

Commit

Permalink
make style
Browse files Browse the repository at this point in the history
  • Loading branch information
NathanHB committed Jan 30, 2024
1 parent 561ec95 commit d34ae2f
Showing 1 changed file with 10 additions and 10 deletions.
20 changes: 10 additions & 10 deletions src/lighteval/logging/info_loggers.py
Original file line number Diff line number Diff line change
Expand Up @@ -101,16 +101,16 @@ def log_args_info(
self.config = config

def log_model_info(self, model_info: ModelInfo) -> None:
"""
Logs the model information.
Args:
model_info (ModelInfo): The model information to be logged.
"""
self.model_name = model_info.model_name
self.model_sha = model_info.model_sha
self.model_dtype = model_info.model_dtype
self.model_size = model_info.model_size
"""
Logs the model information.
Args:
model_info (ModelInfo): The model information to be logged.
"""
self.model_name = model_info.model_name
self.model_sha = model_info.model_sha
self.model_dtype = model_info.model_dtype
self.model_size = model_info.model_size

def log_end_time(self) -> None:
self.end_time = time.perf_counter()
Expand Down

0 comments on commit d34ae2f

Please sign in to comment.