add python-torchviz-git

2 files changed, 51 insertions, 0 deletions

diff --git a/.SRCINFO b/.SRCINFO
new file mode 100644
index 000000000000..38f18aaaf24b
--- /dev/null
+++ b/.SRCINFO
@@ -0,0 +1,18 @@
+pkgbase = python-torchviz-git
+	pkgdesc = A small package to create visualizations of PyTorch execution graphs and traces
+	pkgver = 0.0.1.r13.a02b4a2
+	pkgrel = 1
+	url = https://github.com/szagoruyko/pytorchviz
+	arch = any
+	license = custom
+	makedepends = git
+	makedepends = python-setuptools
+	depends = python-graphviz
+	depends = python-pytorch
+	provides = python-torchviz=0.0.1.r13.a02b4a2
+	conflicts = python-torchviz
+	source = python-torchviz-git::git+https://github.com/szagoruyko/pytorchviz.git
+	md5sums = SKIP
+
+pkgname = python-torchviz-git
+
diff --git a/PKGBUILD b/PKGBUILD
new file mode 100644
index 000000000000..5ab63ff463ed
--- /dev/null
+++ b/PKGBUILD
@@ -0,0 +1,33 @@
+# Maintainer: Butui Hu <hot123tea123@gmail.com>
+
+pkgname=python-torchviz-git
+_pkgver=0.0.1
+pkgver=0.0.1.r13.a02b4a2
+pkgrel=1
+pkgdesc="A small package to create visualizations of PyTorch execution graphs and traces"
+arch=(any)
+url="https://github.com/szagoruyko/pytorchviz"
+license=(custom)
+depends=(python-graphviz python-pytorch)
+makedepends=(git python-setuptools)
+provides=(python-torchviz=${pkgver})
+conflicts=(python-torchviz)
+source=("${pkgname}::git+https://github.com/szagoruyko/pytorchviz.git")
+md5sums=('SKIP')
+
+pkgver() {
+  cd "${pkgname}"
+  ver=$(printf "r%s.%s" "$(git rev-list --count HEAD)" "$(git rev-parse --short HEAD)")
+  echo "${_pkgver}.${ver}"
+}
+
+build() {
+	cd "${srcdir}/${pkgname}"
+	python setup.py build
+}
+
+package() {
+	cd "${srcdir}/${pkgname}"
+	python setup.py install --root="${pkgdir}" --optimize=1 --skip-build
+}
+# vim:set ts=2 sw=2 et: