summary refs log tree commit diff
diff options
context:
space:
mode:
authorTimothee Mathieu <timothee.mathieu@inria.fr>2024-02-12 15:10:07 +0100
committerRicardo Wurmus <rekado@elephly.net>2024-02-16 08:01:24 +0100
commitc83d67b8099c77a9c9afcf4b6960bf27e225cd66 (patch)
treedb0501e1613ce6a188b763aead944eb25c478851
parentcd225af6e6df2aa01b5c0d4b0f895efdc5cdd95e (diff)
downloadguix-c83d67b8099c77a9c9afcf4b6960bf27e225cd66.tar.gz
gnu: Add python-gymnasium.
* gnu/packages/machine-learning.scm (python-gymnasium): New variable.

Change-Id: I22564c63b24ae98896f2e7b7da679463bd6b8496
Signed-off-by: Ricardo Wurmus <rekado@elephly.net>
-rw-r--r--gnu/packages/machine-learning.scm32
1 files changed, 32 insertions, 0 deletions
diff --git a/gnu/packages/machine-learning.scm b/gnu/packages/machine-learning.scm
index 2e20872555..6e94e21f3e 100644
--- a/gnu/packages/machine-learning.scm
+++ b/gnu/packages/machine-learning.scm
@@ -25,6 +25,7 @@
 ;;; Copyright © 2023 Troy Figiel <troy@troyfigiel.com>
 ;;; Copyright © 2024 Sharlatan Hellseher <sharlatanus@gmail.com>
 ;;; Copyright © 2024 David Pflug <david@pflug.io>
+;;; Copyright © 2024 Timothee Mathieu <timothee.mathieu@inria.fr>
 ;;;
 ;;; This file is part of GNU Guix.
 ;;;
@@ -5386,3 +5387,34 @@ performance library of basic building blocks for deep learning applications.")
     (synopsis "Read and write ML models in GGUF for GGML")
     (description "A Python library for reading and writing GGUF & GGML format ML models.")
     (license license:expat)))
+
+(define-public python-gymnasium
+  (package
+    (name "python-gymnasium")
+    (version "0.29.1")
+    (source
+     (origin
+       (method url-fetch)
+       (uri (pypi-uri "gymnasium" version))
+       (sha256
+        (base32 "1cab4wsnlsxn2z90qmymv8ppmsq8yq2amiqwid3r0xfbxx92flqs"))))
+    (build-system pyproject-build-system)
+    (propagated-inputs (list python-cloudpickle python-farama-notifications
+                             python-importlib-metadata python-numpy
+                             python-typing-extensions))
+    (native-inputs (list python-pytest python-scipy))
+    (arguments
+     (list
+      #:phases
+      #~(modify-phases %standard-phases
+          (add-after 'unpack 'create-tests-module
+            (lambda _
+              (with-output-to-file "tests/__init__.py"
+                (lambda _ (display ""))))))))
+    (home-page "https://gymnasium.farama.org/")
+    (synopsis
+     "Standard API for reinforcement learning and a set of reference environments")
+    (description
+     "This package provides a standard API for reinforcement learning and a
+diverse set of reference environments (formerly Gym).")
+    (license license:expat)))