diff --git a/pkgs/development/python-modules/gymnasium/default.nix b/pkgs/development/python-modules/gymnasium/default.nix index 28037b9a6e66..c70a4081cebb 100644 --- a/pkgs/development/python-modules/gymnasium/default.nix +++ b/pkgs/development/python-modules/gymnasium/default.nix @@ -32,7 +32,7 @@ buildPythonPackage rec { pname = "gymnasium"; - version = "1.0.0"; + version = "1.1.0"; pyproject = true; @@ -40,7 +40,7 @@ buildPythonPackage rec { owner = "Farama-Foundation"; repo = "gymnasium"; tag = "v${version}"; - hash = "sha256-Qchuz08yJ0giVrtKLC9vBgr28JrHQyAOCuoS239ivVw="; + hash = "sha256-o+rxEMxiw4BP1lTk0xTXjqXS2C9vuHBzE8F7ZEVHmIc="; }; build-system = [ setuptools ]; diff --git a/pkgs/development/python-modules/stable-baselines3/default.nix b/pkgs/development/python-modules/stable-baselines3/default.nix index 0cca87dd9857..2963770bf252 100644 --- a/pkgs/development/python-modules/stable-baselines3/default.nix +++ b/pkgs/development/python-modules/stable-baselines3/default.nix @@ -34,6 +34,10 @@ buildPythonPackage rec { build-system = [ setuptools ]; + pythonRelaxDeps = [ + "gymnasium" + ]; + dependencies = [ cloudpickle gymnasium diff --git a/pkgs/development/python-modules/tianshou/default.nix b/pkgs/development/python-modules/tianshou/default.nix index 9029669bcd05..9078317fc055 100644 --- a/pkgs/development/python-modules/tianshou/default.nix +++ b/pkgs/development/python-modules/tianshou/default.nix @@ -1,39 +1,48 @@ { lib, - ale-py, + stdenv, buildPythonPackage, - cython, - deepdiff, - docstring-parser, + pythonOlder, fetchFromGitHub, + + # build-system + poetry-core, + + # dependencies + deepdiff, gymnasium, h5py, - imageio, - joblib, - jsonargparse, matplotlib, - mujoco, numba, numpy, - opencv, overrides, packaging, pandas, pettingzoo, - poetry-core, - pybox2d, - pybullet, - pygame, - pymunk, - pytestCheckHook, - pythonOlder, - scipy, sensai-utils, - shimmy, - swig, tensorboard, torch, tqdm, + + # optional-dependencies + docstring-parser, + jsonargparse, + ale-py, + opencv, + shimmy, + pybox2d, + pygame, + swig, + mujoco, + imageio, + cython, + pybullet, + joblib, + scipy, + + # tests + pymunk, + pytestCheckHook, }: buildPythonPackage rec { @@ -156,20 +165,26 @@ buildPythonPackage rec { "test/offline" ]; - disabledTests = [ - # AttributeError: 'TimeLimit' object has no attribute 'test_attribute' - "test_attr_unwrapped" - # Failed: DID NOT RAISE - "test_batch" - # Failed: Raised AssertionError - "test_vecenv" - ]; + disabledTests = + [ + # AttributeError: 'TimeLimit' object has no attribute 'test_attribute' + "test_attr_unwrapped" + # Failed: DID NOT RAISE + "test_batch" + # Failed: Raised AssertionError + "test_vecenv" + ] + ++ lib.optionals stdenv.hostPlatform.isDarwin [ + # Fatal Python error: Aborted + # pettingzoo/classic/tictactoe/tictactoe.py", line 254 in reset + "test_tic_tac_toe" + ]; - meta = with lib; { + meta = { description = "Elegant PyTorch deep reinforcement learning library"; homepage = "https://github.com/thu-ml/tianshou"; changelog = "https://github.com/thu-ml/tianshou/releases/tag/v${version}"; - license = licenses.mit; - maintainers = with maintainers; [ derdennisop ]; + license = lib.licenses.mit; + maintainers = with lib.maintainers; [ derdennisop ]; }; }