summarylogtreecommitdiffstats
path: root/PKGBUILD
blob: 0c52589d81c2c04b405a187a76265c276f7451ea (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
# Maintainer: devome <evinedeng@hotmail.com>

_pkgname=colbert-ai
_pipname="${_pkgname//-/_}"
pkgname="python-${_pkgname}"
pkgver=0.2.21
pkgrel=1
pkgdesc="Efficient and Effective Passage Search via Contextualized Late Interaction over BERT"
arch=("any")
url="https://github.com/stanford-futuredata/ColBERT"
license=('MIT')
depends=("python-bitarray" "python-datasets" "python-flask" "python-git-python" "python-dotenv" "python-ninja" "python-scipy" "python-tqdm" "python-transformers" "python-ujson")
makedepends=('python-build' 'python-installer' 'python-setuptools' 'python-wheel')
optdepends=("python-faiss" "python-pytorch")
source=("${_pkgname}-${pkgver}.tar.gz::https://files.pythonhosted.org/packages/source/${_pipname::1}/${_pipname}/${_pipname}-${pkgver}.tar.gz")
sha256sums=('a8d6fdb4e2272f2b08ed37f8e5096072160d8415d1e40585751898b77e625bab')

build() {
    cd "${_pipname}-${pkgver}"
    python -m build --wheel --no-isolation
}

package() {
    cd "${_pipname}-${pkgver}"
    python -m installer --destdir="${pkgdir}" dist/*.whl
    install -Dm644 LICENSE "${pkgdir}/usr/share/licenses/${pkgname}/LICENSE"
}