Bump

2 files changed, 33 insertions, 10 deletions

diff --git a/.SRCINFO b/.SRCINFO
index 61fe0c55e6b2..7c1f27f4900f 100644
--- a/.SRCINFO
+++ b/.SRCINFO
@@ -1,17 +1,22 @@
 pkgbase = python-tensorboardx-git
 	pkgdesc = Tensorboard for PyTorch
-	pkgver = 1.2.r91.g4093ead
-	pkgrel = 2
+	pkgver = 2.5.1.r36.g092fcb0
+	pkgrel = 1
 	url = https://github.com/lanpa/tensorboardX
 	arch = any
 	license = MIT
-	makedepends = python-setuptools
+	makedepends = git
+	makedepends = python-build
+	makedepends = python-installer
+	makedepends = python-setuptools-scm
+	makedepends = python-wheel
+	depends = python-numpy
 	depends = python-protobuf
-	depends = python-six
+	optdepends = python-crc32c: speed up
+	optdepends = python-soundfile: 200x speed up the add_audio() function
 	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
index 083b3b606049..1bf2b26269ce 100644
--- a/PKGBUILD
+++ b/PKGBUILD
@@ -1,16 +1,29 @@
 # Maintainer of this PKGBUILD file: Martino Pilia <martino.pilia@gmail.com>
 _name=tensorboardX
 pkgname=python-tensorboardx-git
-pkgver=1.2.r91.g4093ead
-pkgrel=2
+pkgver=2.5.1.r36.g092fcb0
+pkgrel=1
 pkgdesc="Tensorboard for PyTorch"
 arch=('any')
 url="https://github.com/lanpa/tensorboardX"
 license=('MIT')
-depends=('python-protobuf' 'python-six')
+depends=(
+    'python-numpy'
+    'python-protobuf'
+    )
+makedepends=(
+    'git'
+    'python-build'
+    'python-installer'
+    'python-setuptools-scm'
+    'python-wheel'
+    )
+optdepends=(
+    'python-crc32c: speed up'
+    'python-soundfile: 200x speed up the add_audio() function'
+    )
 provides=('python-tensorboardx')
 conflicts=('python-tensorboardx')
-makedepends=('python-setuptools')
 source+=("git+https://github.com/lanpa/tensorboardX.git")
 sha256sums=('SKIP')
 
@@ -20,8 +33,13 @@ pkgver() {
 	  sed 's/^v//;s/_/./;s/\([^-]*-g\)/r\1/;s/-/./g'
 }
 
+build() {
+	cd "$srcdir/$_name"
+    python -m build --wheel --no-isolation
+}
+
 package() {
 	cd "$srcdir/$_name"
+    python -m installer --destdir="$pkgdir" dist/*.whl
 	install -D -m644 LICENSE "${pkgdir}/usr/share/licenses/${pkgname}/LICENSE"
-	python setup.py install --optimize=1 --root="$pkgdir"
 }