Emil
5b2d5b5f6e
coordinator / test (push) Canceled after 0s
python / test (push) Canceled after 0s
release / binaries (amd64, darwin) (push) Canceled after 0s
release / binaries (amd64, linux) (push) Canceled after 0s
release / binaries (amd64, windows) (push) Canceled after 0s
release / binaries (arm64, darwin) (push) Canceled after 0s
release / binaries (arm64, linux) (push) Canceled after 0s
release / binaries (arm64, windows) (push) Canceled after 0s
release / wheel (push) Canceled after 0s
release / image (push) Canceled after 0s
users / test (push) Canceled after 0s
release / release (push) Canceled after 0s
29 lines
748 B
Python
29 lines
748 B
Python
"""SDK-built ``similarity-search-parallel`` workload.
|
|
|
|
Same contract as ``similarity-search`` with a per-shard thread pool. See
|
|
``core.py`` for the parallel scoring core and ``definition.py`` for the
|
|
manifest-backed handlers.
|
|
"""
|
|
|
|
from .core import (
|
|
run_search_shard_parallel,
|
|
search_similar_parallel,
|
|
write_search_shards,
|
|
)
|
|
from .definition import (
|
|
MAP_ENTRY_POINT,
|
|
SimilaritySearchParallelSDKWorkload,
|
|
similarity_search_parallel_sdk_definition,
|
|
workload_definition,
|
|
)
|
|
|
|
__all__ = [
|
|
"MAP_ENTRY_POINT",
|
|
"SimilaritySearchParallelSDKWorkload",
|
|
"similarity_search_parallel_sdk_definition",
|
|
"workload_definition",
|
|
"run_search_shard_parallel",
|
|
"search_similar_parallel",
|
|
"write_search_shards",
|
|
]
|