Skip to content

Commit

Permalink
style: ran linting
Browse files Browse the repository at this point in the history
  • Loading branch information
KennethEnevoldsen committed Jan 25, 2024
1 parent d9a13cb commit 05b6bf9
Show file tree
Hide file tree
Showing 3 changed files with 3 additions and 13 deletions.
9 changes: 1 addition & 8 deletions src/seb/benchmark.py
Original file line number Diff line number Diff line change
Expand Up @@ -212,13 +212,6 @@ def evaluate_models(
for model in pbar:
pbar.set_description(f"Running {model.meta.name}")
results.append(
self.evaluate_model(
model,
use_cache=use_cache,
run_model=run_model,
raise_errors=raise_errors,
cache_dir=cache_dir,
verbose=verbose
),
self.evaluate_model(model, use_cache=use_cache, run_model=run_model, raise_errors=raise_errors, cache_dir=cache_dir, verbose=verbose),
)
return results
6 changes: 1 addition & 5 deletions src/seb/registered_tasks/mteb_tasks.py
Original file line number Diff line number Diff line change
Expand Up @@ -103,7 +103,6 @@ def description(self) -> dict[str, Any]:
}



class SwednSummarizationSTS(AbsTaskSTS):
def load_data(self, **kwargs: dict): # noqa: ARG002
"""
Expand Down Expand Up @@ -249,9 +248,7 @@ def dataset_transform(self) -> None:
n += 1
cor_n = text2id[art]

self.relevant_docs[split][str(q_n)] = {
str(text2id[art]): 1, str(text2id[summ]): 1
} # only two correct matches
self.relevant_docs[split][str(q_n)] = {str(text2id[art]): 1, str(text2id[summ]): 1} # only two correct matches


class NorwegianCourtsBitextMining(AbsTaskBitextMining):
Expand Down Expand Up @@ -291,4 +288,3 @@ def dataset_transform(self) -> None:
# Convert to standard format
self.dataset = self.dataset.rename_column("nb", "sentence1")
self.dataset = self.dataset.rename_column("nn", "sentence2")

1 change: 1 addition & 0 deletions src/seb/registered_tasks/swedish.py
Original file line number Diff line number Diff line change
Expand Up @@ -57,6 +57,7 @@ def create_swedn_retrieval() -> Task:
task.domain = ["non-fiction", "news"]
return task


@tasks.register("SwednClustering")
def create_swedn_clustering() -> Task:
from seb.registered_tasks.mteb_tasks_clustering import SwednClustering
Expand Down

0 comments on commit 05b6bf9

Please sign in to comment.