python310Packages.gymnasium: 0.29.0 -> 0.29.1

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

@@ -10,34 +10,67 @@
 , farama-notifications
 , importlib-metadata
 , pythonOlder
+, ffmpeg
+, jax
+, jaxlib
+, matplotlib
+, moviepy
+, opencv4
+, pybox2d
+, pygame
+, pytestCheckHook
+, scipy
 }:
 
 buildPythonPackage rec {
   pname = "gymnasium";
-  version = "0.29.0";
+  version = "0.29.1";
+  format = "pyproject";
 
   src = fetchFromGitHub {
     owner = "Farama-Foundation";
-    repo = pname;
+    repo = "gymnasium";
     rev = "refs/tags/v${version}";
-    hash = "sha256-4YaEFEWSOTEdGgO1kSOleZQp7OrcOf+WAT/E0BWeoKI=";
+    hash = "sha256-L7fn9FaJzXwQhjDKwI9hlFpbPuQdwynU+Xjd8bbjxiw=";
   };
 
-  format = "pyproject";
-
   nativeBuildInputs = [ setuptools ];
 
   propagatedBuildInputs = [
-    jax-jumpy
     cloudpickle
-    numpy
-    gym-notices
-    typing-extensions
     farama-notifications
+    gym-notices
+    jax-jumpy
+    numpy
+    typing-extensions
   ] ++ lib.optionals (pythonOlder "3.10") [ importlib-metadata ];
 
   pythonImportsCheck = [ "gymnasium" ];
 
+  nativeCheckInputs = [
+    ffmpeg
+    jax
+    jaxlib
+    matplotlib
+    moviepy
+    opencv4
+    pybox2d
+    pygame
+    pytestCheckHook
+    scipy
+  ];
+
+  disabledTestPaths = [
+    # mujoco is required for those tests but the mujoco python bindings are not packaged in nixpkgs.
+    "tests/envs/mujoco/test_mujoco_custom_env.py"
+
+    # Those tests need to write on the filesystem which cause them to fail.
+    "tests/experimental/wrappers/test_record_video.py"
+    "tests/utils/test_save_video.py"
+    "tests/wrappers/test_record_video.py"
+    "tests/wrappers/test_video_recorder.py"
+  ];
+
   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";