summarylogtreecommitdiffstats
path: root/PKGBUILD
diff options
context:
space:
mode:
Diffstat (limited to 'PKGBUILD')
-rw-r--r--PKGBUILD27
1 files changed, 21 insertions, 6 deletions
diff --git a/PKGBUILD b/PKGBUILD
index a6929c329b9c..aa50a1f6995f 100644
--- a/PKGBUILD
+++ b/PKGBUILD
@@ -1,13 +1,28 @@
# Maintainer: Butui Hu <hot123tea123@gmail.com>
pkgname=python-pretrainedmodels-git
pkgver=r154.8aae3d8
-pkgrel=3
+pkgrel=5
pkgdesc="Pretrained ConvNets for PyTorch"
arch=(any)
url="https://github.com/Cadene/pretrained-models.pytorch"
-license=('BSD')
-depends=(python-munch python-pytorch python-scipy python-torchvision python-tqdm )
-makedepends=(git python-setuptools)
+license=('BSD-3-Clause')
+depends=(
+ python-munch
+ python-numpy
+ python-pillow
+ python-pytorch
+ python-scipy
+ python-six
+ python-torchvision
+ python-tqdm
+)
+makedepends=(
+ git
+ python-build
+ python-installer
+ python-setuptools
+ python-wheel
+)
provides=(python-pretrainedmodels)
conflicts=(python-pretrainedmodels)
source=("${pkgname}::git+https://github.com/Cadene/pretrained-models.pytorch.git")
@@ -20,12 +35,12 @@ pkgver() {
build() {
cd "${pkgname}"
- python setup.py build
+ python -m build --wheel --no-isolation
}
package() {
cd "${pkgname}"
- python setup.py install --root="${pkgdir}" --optimize=1 --skip-build
+ python -m installer --destdir="${pkgdir}" dist/*.whl
install -Dm644 LICENSE.txt -t "${pkgdir}/usr/share/licenses/${pkgname}"
}
# vim:set ts=2 sw=2 et: