Skip to content

Commit

Permalink
typing error
Browse files Browse the repository at this point in the history
  • Loading branch information
linogaliana committed Aug 12, 2024
1 parent bb5a4bc commit 18c5f97
Show file tree
Hide file tree
Showing 4 changed files with 7 additions and 5 deletions.
2 changes: 1 addition & 1 deletion src/evaluation/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@
from .reranking_perf import compare_performance_reranking
from .retrieval_evaluation_measures import RetrievalEvaluationMeasure
from .retrieval_evaluator import RetrievalEvaluator
from .utils import build_chain_reranker_test, choosing_reranker_test, use_sbert_retrieval_evaluator, hist_results, plot_results
from .utils import build_chain_reranker_test, choosing_reranker_test, hist_results, plot_results, use_sbert_retrieval_evaluator

__all__ = [
"RetrievalConfiguration",
Expand Down
4 changes: 2 additions & 2 deletions src/evaluation/retrieval_evaluation_measures.py
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
# retrieval_evaluation_measures.py

import numpy as np
from abc import ABC
from dataclasses import dataclass, field


class RetrievalEvaluationMeasure:
## Measures #############
Expand Down
4 changes: 2 additions & 2 deletions src/evaluation/retrieval_evaluator.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,7 @@
from tqdm import tqdm

from src.config import EMB_DEVICE, EMB_MODEL_NAME
from src.db_building import build_vector_database, reload_database_from_local_dir
from src.db_building import build_vector_database

from .eval_configuration import RetrievalConfiguration
from .retrieval_evaluation_measures import RetrievalEvaluationMeasure
Expand All @@ -26,7 +26,7 @@ def _build_vector_database(path_data: str, config: RetrievalConfiguration) -> Ch
"""
Building vector database based on a given embedding model
"""
from db_building import build_database_from_dataframe, reload_database_from_local_dir
from db_building import reload_database_from_local_dir

embedding_model_name = config.get("embedding_model_name", EMB_MODEL_NAME)
persist_directory = "./data/chroma_db"
Expand Down
2 changes: 2 additions & 0 deletions src/reranking/reranking_functions.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,5 @@
from collections.abc import Any, Sequence

import numpy as np
import torch
import torch.nn.functional as F
Expand Down

0 comments on commit 18c5f97

Please sign in to comment.