Python ChunkArray Examples

Example #1

 def do_something(self, docs, **kwargs):
     for doc in docs:
         chunks = ChunkArray(
             list(filter(lambda d: d.modality == self._mode, doc.chunks)),
             doc)
         assert chunks[0].content == self._mode
         assert len(chunks) == 1
         doc.chunks = chunks

Example #2

File: test_docs_matrix_tail_pea.py Project: paddlelaw/jina

 def do_something(self, docs, **kwargs):
     for doc in docs:
         if len(doc.chunks) > 0:
             chunks = ChunkArray(
                 (d for d in doc.chunks if d.modality == self._mode), doc
             )
             assert chunks[0].content == self._mode
             assert len(chunks) == 1
             doc.chunks = chunks

Example #3

File: test_chunkarray.py Project: paddlelaw/jina

def chunkarray(chunks, reference_doc):
    return ChunkArray(doc_views=chunks, reference_doc=reference_doc)

Example #4

from jina.types.arrays.chunk import ChunkArray
from jina.types.arrays.match import MatchArray
from jina.types.ndarray.generic import NdArray
from jina.types.request import Request
from jina.types.score import NamedScore


@pytest.mark.parametrize(
    'obj',
    [
        Document(),
        Request(),
        NamedScore(),
        NdArray(),
        MatchArray([Document()], Document()),
        ChunkArray([Document()], Document()),
    ],
)
def test_builtin_str_repr_no_content(obj):
    print(obj)
    print(f'{obj!r}')


@pytest.mark.parametrize(
    'obj',
    [
        Document(content='123', chunks=[Document(content='abc')]),
        NamedScore(
            op_name='operation',
            value=10.0,
            ref_id='10' * 16,

Example #5

def chunkarray(chunks, reference_doc):
    return ChunkArray(docs_proto=chunks, reference_doc=reference_doc)