summarylogtreecommitdiffstats
path: root/PKGBUILD
blob: b2d63a8440cced46f350f166c8e77350929d5d21 (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
29
30
31
32
33
34
# Maintainer: piernov <piernov@piernov.org>

pkgname=python-qualia-core
pkgver=2.2.0
pkgrel=1
pkgdesc="End-to-end training, quantization and deployment framework for deep neural networks on microcontrollers."
arch=('any')
url="https://github.com/LEAT-EDGE/qualia-core"
license=('AGPL-3.0')
depends=('python' 'python-numpy' 'python-tomlkit' 'python-colorful' 'python-gitpython' 'python-blosc2' 'python-pydantic' 'python-typing_extensions')
makedepends=('git' 'python-build' 'python-installer' 'python-wheel' 'python-pdm-backend')
optdepends=('python-qualia-codegen-core: support for deployment using Qualia-CodeGen C code generation tool'
            'python-keras: support for training and deployment using Keras backend'
            'python-pytorch-lightning: support for training, quantization and deployment using PyTorch backend'
            'python-imageio: support for GTSRB dataset'
            'python-scikit-image: support for GTSRB dataset'
            'python-torchaudio: support for Google Speech Commands dataset'
            'python-pytorch3d: support for 3D rotation data augmentation'
            'python-torchvision: support for image data augmentations'
            'python-clearml: support for ClearML experiment tracking'
            'python-tensorboard: support for TensorBoard and ClearML experiment tracking'
            'python-pyserial: support for on-target evaluation over (virtual) serial port')
source=("git+https://github.com/leat-edge/qualia-core.git#tag=$pkgver")
md5sums=('SKIP')

build() {
  cd "$srcdir"/qualia-core
  python -m build --wheel --no-isolation
}

package() {
  cd "$srcdir"/qualia-core
  python -m installer --destdir="$pkgdir" dist/*.whl
}