diff options
author | Benoît Allard | 2021-03-07 16:09:12 +0100 |
---|---|---|
committer | Benoît Allard | 2021-03-07 16:09:12 +0100 |
commit | 34ce42322182be6b011c3b444e6ce4278df0b931 (patch) | |
tree | 6b06faa47cdcb19de257d65df68694b3930e0715 | |
download | aur-34ce42322182be6b011c3b444e6ce4278df0b931.tar.gz |
stable-baselines3-1.0rc1
-rw-r--r-- | .SRCINFO | 24 | ||||
-rw-r--r-- | PKGBUILD | 28 |
2 files changed, 52 insertions, 0 deletions
diff --git a/.SRCINFO b/.SRCINFO new file mode 100644 index 000000000000..c9d80b44ba74 --- /dev/null +++ b/.SRCINFO @@ -0,0 +1,24 @@ +pkgbase = python-stable-baselines3 + pkgdesc = A set of reliable implementations of reinforcement learning algorithms in PyTorch + pkgver = 1.0rc1 + pkgrel = 1 + url = https://github.com/DLR-RM/stable-baselines3 + arch = any + license = MIT + depends = python + depends = python-gym + depends = python-numpy + depends = python-pytorch + depends = python-cloudpickle + depends = python-pandas + depends = python-matplotlib + optdepends = python-opencv: For render + optdepends = python-atari-py: For atari games + optdepends = python-pillow: For atari games + optdepends = tensorboard: Tensorboard support + optdepends = python-psutils: Checking memory taken by replay buffer + source = python-stable-baselines3-1.0rc1.tar.gz::https://github.com/DLR-RM/stable-baselines3/archive/v1.0rc1.tar.gz + md5sums = 6a644fd4d49d6249cccfb3c09aecc05e + +pkgname = python-stable-baselines3 + diff --git a/PKGBUILD b/PKGBUILD new file mode 100644 index 000000000000..e09bcd967d32 --- /dev/null +++ b/PKGBUILD @@ -0,0 +1,28 @@ +# Maintainer: Benoît Allard <benoit.allard@gnx.de> +pkgname=python-stable-baselines3 +pkgver=1.0rc1 +pkgrel=1 +pkgdesc="A set of reliable implementations of reinforcement learning algorithms in PyTorch" +arch=('any') +url="https://github.com/DLR-RM/stable-baselines3" +license=('MIT') +depends=("python" "python-gym" "python-numpy" "python-pytorch" "python-cloudpickle" "python-pandas" "python-matplotlib") +optdepends=('python-opencv: For render' + 'python-atari-py: For atari games' + 'python-pillow: For atari games' + 'tensorboard: Tensorboard support' + 'python-psutils: Checking memory taken by replay buffer') +_name=${pkgname#python-} +source=("$pkgname-$pkgver.tar.gz::https://github.com/DLR-RM/$_name/archive/v$pkgver.tar.gz") +md5sums=('6a644fd4d49d6249cccfb3c09aecc05e') + +build() { + cd $_name-$pkgver + export PYTHONSEED=1 + python setup.py build +} + +package() { + cd $_name-$pkgver + python setup.py install --root="$pkgdir" --optimize=1 --skip-build +} |