summarylogtreecommitdiffstats
path: root/PKGBUILD
blob: 742083599f61a5d73ec68136b36f411bcdabaef5 (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
28
# shellcheck shell=bash disable=SC2034,SC2154
# Maintainer: Mateen Ulhaq <mulhaq2005+aur at gmail dot com>

# https://aur.archlinux.org/packages/updaurpkg-git
# $ updaurpkg --apply

_source_type=pypi-releases
_pypi_package=torchinfo
_upstreamver='1.8.0'
_pkgname=${_pypi_package}
_repo=tyleryep/torchinfo

pkgname=python-$_pkgname
pkgver=${_upstreamver}
pkgrel=1
pkgdesc="Model summary in PyTorch, based off of the original torchsummary"
arch=(x86_64)
url=https://github.com/$_repo
depends=(python-pytorch python-torchvision)
makedepends=(python-setuptools)
license=(MIT)
source=("https://files.pythonhosted.org/packages/source/${_pkgname:0:1}/$_pkgname/${_pkgname//-/_}-$pkgver.tar.gz")
sha256sums=('72e94b0e9a3e64dc583a8e5b7940b8938a1ac0f033f795457f27e6f4e7afa2e9')

package() {
  cd "$srcdir/$_pkgname-$pkgver" || return 1
  python setup.py install --root="$pkgdir" --optimize=1 || return 1
}