package submission

2 files changed, 48 insertions, 0 deletions

diff --git a/.SRCINFO b/.SRCINFO
new file mode 100644
index 000000000000..08d0f59a0a59
--- /dev/null
+++ b/.SRCINFO
@@ -0,0 +1,20 @@
+pkgbase = python-tensorboardx-git
+	pkgdesc = Tensorboard for PyTorch
+	pkgver = 1.2.r90.gddca1a9
+	pkgrel = 1
+	url = https://github.com/lanpa/tensorboardX
+	arch = any
+	license = MIT
+	makedepends = python-setuptools
+	depends = python-pytorch
+	depends = python-torchvision
+	depends = python-protobuf
+	depends = python-six
+	optdepends = tensorboard: to visualise TensorBoard data
+	provides = python-tensorboardx
+	conflicts = python-tensorboardx
+	source = git+https://github.com/lanpa/tensorboardX.git
+	sha256sums = SKIP
+
+pkgname = python-tensorboardx-git
+
diff --git a/PKGBUILD b/PKGBUILD
new file mode 100644
index 000000000000..a9b0f8337879
--- /dev/null
+++ b/PKGBUILD
@@ -0,0 +1,28 @@
+# Maintainer of this PKGBUILD file: Martino Pilia <martino.pilia@gmail.com>
+_name=tensorboardX
+pkgname=python-tensorboardx-git
+pkgver=1.2.r90.gddca1a9
+pkgrel=1
+pkgdesc="Tensorboard for PyTorch"
+arch=('any')
+url="https://github.com/lanpa/tensorboardX"
+license=('MIT')
+depends=('python-pytorch' 'python-torchvision' 'python-protobuf' 'python-six')
+optdepends=('tensorboard: to visualise TensorBoard data')
+provides=('python-tensorboardx')
+conflicts=('python-tensorboardx')
+makedepends=('python-setuptools')
+source+=("git+https://github.com/lanpa/tensorboardX.git")
+sha256sums=('SKIP')
+
+pkgver() {
+  cd "$_name"
+  git describe --long --tags | \
+	  sed 's/^v//;s/_/./;s/\([^-]*-g\)/r\1/;s/-/./g'
+}
+
+package() {
+	cd "$srcdir/$_name"
+	install -D -m644 LICENSE "${pkgdir}/usr/share/licenses/${pkgname}/LICENSE"
+	python setup.py install --optimize=1 --root="$pkgdir"
+}