1 files changed, 5 insertions, 10 deletions
diff --git a/PKGBUILD b/PKGBUILD
index 469ee300c5f3..05358c370eb0 100644
--- a/PKGBUILD
+++ b/PKGBUILD
@@ -2,29 +2,24 @@
 pkgname=python-keras-flops
 _name=${pkgname#python-}
 pkgver=0.1.2
-pkgrel=1
+pkgrel=2
 pkgdesc="FLOPs calculator for TF2+"
 arch=('any')
 url="https://github.com/tokusumi/keras-flops"
 license=('MIT')
-depends=('python-tensorflow')
-makedepends=('python-setuptools' 'python-dephell')
+depends=('python-tensorflow' 'python-numpy')
+makedepends=('python-installer' 'python-poetry')
 options=()
 source=("https://files.pythonhosted.org/packages/source/${_name::1}/$_name/$_name-$pkgver.tar.gz")
 sha256sums=('2561043d435dce3b0b6a68573a538facddbcfa637faf7ae7f48b2e8f373ed84e')
 
-prepare() {
-	cd "$srcdir/${_name}-${pkgver}"
-    dephell deps convert --from pyproject.toml --to setup.py
-}
-
 build() {
 	cd "$srcdir/${_name}-${pkgver}"
-    python setup.py build
+    poetry build --no-interaction --format wheel
 }
 
 package() {
 	cd "$srcdir/${_name}-${pkgver}"
-    PYTHONHASHSEED=0 python setup.py install --root="$pkgdir" --optimize=1 --skip-build
+    python -m installer --destdir="$pkgdir" dist/*.whl
     install -Dm644 LICENSE "$pkgdir/usr/share/licenses/python-keras-flops/LICENSE"
 }