Apply suggestions from code review

Co-authored-by: Nathan Habib <[email protected]> Co-authored-by: Albert Villanova del Moral <[email protected]>
huggingface · Dec 16, 2024 · fb4ecdc · fb4ecdc
1 parent 431b4f2
commit fb4ecdc
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/src/lighteval/main_endpoint.py b/src/lighteval/main_endpoint.py
@@ -98,7 +98,6 @@ def openai(
     """
     from lighteval.logging.evaluation_tracker import EvaluationTracker
 
-    # from lighteval.models.model_input import GenerationParameters
     from lighteval.models.endpoints.openai_model import OpenAIModelConfig
     from lighteval.models.model_input import GenerationParameters
     from lighteval.pipeline import EnvConfig, ParallelismManager, Pipeline, PipelineParameters

diff --git a/src/lighteval/models/transformers/transformers_model.py b/src/lighteval/models/transformers/transformers_model.py
@@ -238,7 +238,7 @@ def get_model_sha(self):
 @dataclass
 class BaseModelConfig(TransformersModelConfig):
     def __post_init__(self):
-        super()
+        super().__post_init__()
 
         logger.warning(
             "Careful, BaseModelConfig is deprecated and will be removed, you should use TransformersModelConfig instead!"