python3Packages.gymnasium: init at 0.28.1

pkgs/development/python-modules/gymnasium/default.nix

@@ -0,0 +1,47 @@
+{ lib
+, buildPythonPackage
+, fetchFromGitHub
+, setuptools
+, numpy
+, cloudpickle
+, gym-notices
+, jax-jumpy
+, typing-extensions
+, farama-notifications
+, importlib-metadata
+, pythonOlder
+}:
+
+buildPythonPackage rec {
+  pname = "gymnasium";
+  version = "0.28.1";
+
+  src = fetchFromGitHub {
+    owner = "Farama-Foundation";
+    repo = pname;
+    rev = "v${version}";
+    hash = "sha256-7rRF21H3IxbgmqxvtC370kr0exLgfg3e2tA3J49xuao=";
+  };
+
+  format = "pyproject";
+
+  nativeBuildInputs = [ setuptools ];
+
+  propagatedBuildInputs = [
+    jax-jumpy
+    cloudpickle
+    numpy
+    gym-notices
+    typing-extensions
+    farama-notifications
+  ] ++ lib.optionals (pythonOlder "3.10") [ importlib-metadata ];
+
+  pythonImportsCheck = [ "gymnasium" ];
+
+  meta = with lib; {
+    description = "A standard API for reinforcement learning and a diverse set of reference environments (formerly Gym)";
+    homepage = "https://github.com/Farama-Foundation/Gymnasium";
+    license = licenses.mit;
+    maintainers = with maintainers; [ GaetanLepage ];
+  };
+}

pkgs/top-level/python-packages.nix

@@ -4272,6 +4272,8 @@ self: super: with self; {
 
   gym-notices = callPackage ../development/python-modules/gym-notices { };
 
+  gymnasium = callPackage ../development/python-modules/gymnasium { };
+
   gyp = callPackage ../development/python-modules/gyp { };
 
   h11 = callPackage ../development/python-modules/h11 { };