diff --git a/README.md b/README.md index df9a75a..7300043 100644 --- a/README.md +++ b/README.md @@ -1,5 +1,5 @@ -[![HF Nuclia](https://img.shields.io/badge/%F0%9F%A4%97-models-yellow)](https://huggingface.co/nuclia) +[![HF Nuclia](https://img.shields.io/badge/%F0%9F%A4%97-nuclia_HF-yellow)](https://huggingface.co/nuclia) [![GitHub - License](https://img.shields.io/github/license/nuclia/nuclia-eval?logo=github&style=flat&color=green)][#github-license] [![PyPI - Python Version](https://img.shields.io/pypi/pyversions/nuclia-eval?logo=pypi&style=flat&color=blue)][#pypi-package] [![PyPI - Package Version](https://img.shields.io/pypi/v/nuclia-eval?logo=pypi&style=flat&color=orange)][#pypi-package] diff --git a/src/nuclia_eval/models/__init__.py b/src/nuclia_eval/models/__init__.py index 64735d3..04dbb54 100644 --- a/src/nuclia_eval/models/__init__.py +++ b/src/nuclia_eval/models/__init__.py @@ -1,4 +1,5 @@ """This module contains the ML models used to evaluate the quality of the RAG experience.""" + from nuclia_eval.models.remi import REMiEvaluator -__all__ = ["REMiEvaluator"] \ No newline at end of file +__all__ = ["REMiEvaluator"] diff --git a/src/nuclia_eval/models/base.py b/src/nuclia_eval/models/base.py index 0505683..3374d21 100644 --- a/src/nuclia_eval/models/base.py +++ b/src/nuclia_eval/models/base.py @@ -4,7 +4,7 @@ from nuclia_eval.metrics.base import DiscreteScoreReasonResponse, DiscreteScoreResponse -class RAGEvaluator(ABC): # pragma: no cover +class RAGEvaluator(ABC): # pragma: no cover """Base class for all RAG evaluators""" @abstractmethod diff --git a/src/nuclia_eval/utils.py b/src/nuclia_eval/utils.py index d6ec117..0cdb26d 100644 --- a/src/nuclia_eval/utils.py +++ b/src/nuclia_eval/utils.py @@ -62,7 +62,7 @@ def load_lora_low_mem( model = model.to(device) -def inherit_docstrings(cls): # pragma: no cover +def inherit_docstrings(cls): # pragma: no cover for name, func in vars(cls).items(): if not func.__doc__ and hasattr(getattr(cls, name), "__doc__"): func.__doc__ = getattr(cls, name).__doc__