Source code for jina.types.score

from typing import Optional, List

from ..mixin import ProtoTypeMixin
from ...excepts import BadNamedScoreType
from ...helper import typename
from ...proto import jina_pb2

__all__ = ['NamedScore']


[docs]class NamedScore(ProtoTypeMixin): """ :class:`NamedScore` is one of the **primitive data type** in Jina. It offers a Pythonic interface to allow users access and manipulate :class:`jina.jina_pb2.NamedScoreProto` object without working with Protobuf itself. To create a :class:`NamedScore` object, simply: .. highlight:: python .. code-block:: python from jina.types.score import NamedScore score = NamedScore() score.value = 10.0 :class:`NamedScore` can be built from ``jina_pb2.NamedScoreProto`` (as a weak reference or a deep copy) or from a set of `attributes` from ``jina_pb2.NamedScoreProto`` passed to the constructor. .. highlight:: python .. code-block:: python from jina.types.score import NamedScore from jina_pb2 import NamedScoreProto score = NamedScore(value=10.0, op_name='ranker', description='score computed by ranker') score_proto = NamedScoreProto() score_proto.value = 10.0 score = NamedScore(score_proto) :param score: The score to construct from, depending on the ``copy``, it builds a view or a copy from it. :type score: Optional[jina_pb2.NamedScoreProto] :param copy: When ``score`` is given as a :class:`NamedScoreProto` object, build a view (i.e. weak reference) from it or a deep copy from it. :type copy: bool :param kwargs: Other parameters to be set """ def __init__( self, score: Optional[jina_pb2.NamedScoreProto] = None, copy: bool = False, **kwargs, ): self._pb_body = jina_pb2.NamedScoreProto() try: if isinstance(score, jina_pb2.NamedScoreProto): if copy: self._pb_body.CopyFrom(score) else: self._pb_body = score elif score is not None: # note ``None`` is not considered as a bad type raise ValueError(f'{typename(score)} is not recognizable') except Exception as ex: raise BadNamedScoreType( f'fail to construct a NamedScore from {score}' ) from ex self.set_attrs(**kwargs) @property def ref_id(self) -> str: """ Return the ``ref_id`` of this NamedScore, the `id` of which this NamedScore is a score. :return:: the ref_id """ return self._pb_body.ref_id @ref_id.setter def ref_id(self, val: str): """ Set the ``ref_id`` to :param: `val`. :param val: The ref_id value to set """ self._pb_body.ref_id = val @property def operands(self) -> List['NamedScore']: """ Returns list of nested NamedScore operands. :return:: list of nested NamedScore operands. """ return [NamedScore(operand) for operand in self._pb_body.operands]
[docs] def set_attrs(self, **kwargs): """Udate Document fields with key-value specified in kwargs. :param kwargs: Key-value parameters to be set """ for k, v in kwargs.items(): if isinstance(v, (list, tuple)): self._pb_body.ClearField(k) if k == 'operands': scores_to_add = [] for item in v: if isinstance(item, NamedScore): score_to_add = item elif isinstance(item, jina_pb2.NamedScoreProto): score_to_add = NamedScore(item) elif isinstance(item, dict): score_to_add = NamedScore(**item) else: raise AttributeError(f'{item} is not recognized.') scores_to_add.append(score_to_add) for score_to_add in scores_to_add: s = self._pb_body.operands.add() s.CopyFrom(score_to_add._pb_body) else: raise AttributeError( f'{k} is not recognized, the only list argument is operands' ) else: if ( hasattr(NamedScore, k) and isinstance(getattr(NamedScore, k), property) and getattr(NamedScore, k).fset ): # if class property has a setter setattr(self, k, v) elif hasattr(self._pb_body, k): # no property setter, but proto has this attribute so fallback to proto setattr(self._pb_body, k, v) else: raise AttributeError(f'{k} is not recognized')