Commit ef6d77c8 authored by Gaetan Lepage's avatar Gaetan Lepage
Browse files

python312Packages.ale-py: 0.10.2 -> 0.11.0 

Diff: https://github.com/Farama-Foundation/Arcade-Learning-Environment/compare/refs/tags/v0.10.2...refs/tags/v0.11.0

Changelog: https://github.com/Farama-Foundation/Arcade-Learning-Environment/releases/tag/v0.11.0
parent a9fb8169

pkgs/development/python-modules/ale-py/default.nix

+41 −14
Original line number Diff line number Diff line
@@ -9,10 +9,10 @@ 
  ninja,

  pybind11,

  setuptools,

  wheel,



  # buildInputs

  SDL2,

  opencv,

  zlib,



  # dependencies
@@ -22,19 +22,25 @@ 


  # tests

  gymnasium,

  opencv-python,

  pytestCheckHook,



  # Whether to enable recently added vector environments:

  # https://github.com/Farama-Foundation/Arcade-Learning-Environment/blob/v0.11.0/docs/vector-environment.md

  # FIXME: Disabled by default as it mysteriously causes stable-baselines3's tests to hang indefinitely.

  withVectorEnv ? false,

}:



buildPythonPackage rec {

  pname = "ale-py";

  version = "0.10.2";

  version = "0.11.0";

  pyproject = true;



  src = fetchFromGitHub {

    owner = "Farama-Foundation";

    repo = "Arcade-Learning-Environment";

    tag = "v${version}";

    hash = "sha256-CGUlQFQoQZqs+Jd3IU/o50VwX+tEHrs3KHrcVWahEpo=";

    hash = "sha256-RgFVpbjJp54FncQzFtdZM7p/1GBMsQ2HvLgIoaokiQc=";

  };



  build-system = [
@@ -42,15 +48,19 @@ buildPythonPackage rec { 
    ninja

    pybind11

    setuptools

    wheel

  ];



  buildInputs = [

  buildInputs =

    [

      SDL2

      zlib

    ]

    ++ lib.optionals withVectorEnv [

      opencv

    ];



  dependencies = [

    gymnasium

    importlib-resources

    numpy

    typing-extensions
@@ -60,16 +70,30 @@ buildPythonPackage rec { 
    # Relax the pybind11 version

    ''

      substituteInPlace src/ale/python/CMakeLists.txt \

        --replace-fail 'find_package(pybind11 ''${PYBIND11_VER} QUIET)' 'find_package(pybind11 QUIET)'

        --replace-fail \

          'find_package(pybind11 ''${PYBIND11_VER} QUIET)' \

          'find_package(pybind11 QUIET)'

    ''

    + lib.optionalString (!withVectorEnv) ''

      substituteInPlace setup.py \

        --replace-fail \

          "-DBUILD_VECTOR_LIB=ON" \

          "-DBUILD_VECTOR_LIB=OFF"

    '';



  dontUseCmakeConfigure = true;



  pythonImportsCheck = [ "ale_py" ];



  nativeCheckInputs = [

  doCheck = false;



  nativeCheckInputs =

    [

      gymnasium

      pytestCheckHook

    ]

    + lib.optionals withVectorEnv [

      opencv-python

    ];



  disabledTests = [
@@ -81,6 +105,8 @@ buildPythonPackage rec { 
    # The user is expected to manually download the roms:

    # https://github.com/Farama-Foundation/Arcade-Learning-Environment/blob/v0.9.0/docs/faq.md#i-downloaded-ale-and-i-installed-it-successfully-but-i-cannot-find-any-rom-file-at-roms-do-i-have-to-get-them-somewhere-else

    "test_check_env"

    "test_reset_step_shapes"

    "test_rollout_consistency"

    "test_sound_obs"

  ];
@@ -92,7 +118,8 @@ buildPythonPackage rec { 
    license = lib.licenses.gpl2;

    maintainers = with lib.maintainers; [ billhuang ];

    badPlatforms = [

      # fails to link with missing library

      # FAILED: src/ale/libale.a

      # /bin/sh: CMAKE_CXX_COMPILER_AR-NOTFOUND: command not found

      lib.systems.inspect.patterns.isDarwin

    ];

  };