run ruff

huggingface · Jul 8, 2024 · baafc49 · baafc49
1 parent 311d68b
commit baafc49
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 6 deletions.
diff --git a/src/lighteval/models/dummy_model.py b/src/lighteval/models/dummy_model.py
@@ -28,10 +28,14 @@
 from transformers import AutoTokenizer
 
 from lighteval.models.abstract_model import LightevalModel
-from lighteval.models.model_config import EnvConfig, DummyModelConfig
-from lighteval.models.model_output import LoglikelihoodSingleTokenReturn, LoglikelihoodReturn, GenerateReturn
-from lighteval.tasks.requests import LoglikelihoodSingleTokenRequest, LoglikelihoodRollingRequest, LoglikelihoodRequest, \
-    GreedyUntilRequest
+from lighteval.models.model_config import DummyModelConfig, EnvConfig
+from lighteval.models.model_output import GenerateReturn, LoglikelihoodReturn, LoglikelihoodSingleTokenReturn
+from lighteval.tasks.requests import (
+    GreedyUntilRequest,
+    LoglikelihoodRequest,
+    LoglikelihoodRollingRequest,
+    LoglikelihoodSingleTokenRequest,
+)
 
 
 class DummyModel(LightevalModel):

diff --git a/src/lighteval/models/model_loader.py b/src/lighteval/models/model_loader.py
@@ -33,11 +33,11 @@
     AdapterModelConfig,
     BaseModelConfig,
     DeltaModelConfig,
+    DummyModelConfig,
     EnvConfig,
     InferenceEndpointModelConfig,
     InferenceModelConfig,
     TGIModelConfig,
-    DummyModelConfig
 )
 from lighteval.models.tgi_model import ModelClient
 from lighteval.utils import NO_TGI_ERROR_MSG, is_accelerate_available, is_tgi_available
@@ -151,4 +151,4 @@ def load_model_with_accelerate_or_default(
 
 
 def load_dummy_model(config: DummyModelConfig, env_config: EnvConfig):
-    return DummyModel(config=config, env_config=env_config), ModelInfo(model_name="dummy", model_sha=str(config.seed))
+    return DummyModel(config=config, env_config=env_config), ModelInfo(model_name="dummy", model_sha=str(config.seed))