summaryrefslogtreecommitdiff
path: root/gnu/packages/machine-learning.scm
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 /gnu/packages/machine-learning.scm
parentcd225af6e6df2aa01b5c0d4b0f895efdc5cdd95e (diff)
gnu: Add python-gymnasium.
* gnu/packages/machine-learning.scm (python-gymnasium): New variable. Change-Id: I22564c63b24ae98896f2e7b7da679463bd6b8496 Signed-off-by: Ricardo Wurmus <rekado@elephly.net>
Diffstat (limited to 'gnu/packages/machine-learning.scm')
-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)))