summarylogtreecommitdiffstats
diff options
context:
space:
mode:
authorBenoît Allard2021-03-07 16:09:12 +0100
committerBenoît Allard2021-03-07 16:09:12 +0100
commit34ce42322182be6b011c3b444e6ce4278df0b931 (patch)
tree6b06faa47cdcb19de257d65df68694b3930e0715
downloadaur-34ce42322182be6b011c3b444e6ce4278df0b931.tar.gz
stable-baselines3-1.0rc1
-rw-r--r--.SRCINFO24
-rw-r--r--PKGBUILD28
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
+}