Something went wrong. Try again.
lol
Something went wrong. Try again.
3.1 kB · 128 lines
Nix
at master
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129{ lib, stdenv, buildPythonPackage, fetchFromGitHub,
# build-system cmake, ninja, pybind11, setuptools,
# buildInputs SDL2, opencv, zlib,
# dependencies importlib-resources, numpy, typing-extensions, jax,
# 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.11.2"; pyproject = true;
src = fetchFromGitHub { owner = "Farama-Foundation"; repo = "Arcade-Learning-Environment"; tag = "v${version}"; hash = "sha256-4IkjW8HX21uBEHFtb3qETxco6FfDMgLbG1BDHWwvn58="; };
build-system = [ cmake ninja pybind11 setuptools ];
buildInputs = [ SDL2 zlib ] ++ lib.optionals withVectorEnv [ opencv ];
dependencies = [ gymnasium importlib-resources numpy typing-extensions ] ++ lib.optionals stdenv.hostPlatform.isLinux [ jax ];
postPatch = # Relax the pybind11 version '' substituteInPlace src/ale/python/CMakeLists.txt \ --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" ];
doCheck = false;
nativeCheckInputs = [ gymnasium pytestCheckHook ] + lib.optionals withVectorEnv [ opencv-python ];
disabledTests = [ # Fatal Python error: Aborted # line 414 in test_display_screen "test_display_screen"
# test_atari_env.py tests fail on the majority of the environments because the ROM are missing. # 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" ];
meta = { description = "Simple framework that allows researchers and hobbyists to develop AI agents for Atari 2600 games"; mainProgram = "ale-import-roms"; homepage = "https://github.com/mgbellemare/Arcade-Learning-Environment"; changelog = "https://github.com/Farama-Foundation/Arcade-Learning-Environment/releases/tag/v${version}"; license = lib.licenses.gpl2; maintainers = with lib.maintainers; [ billhuang ]; badPlatforms = [ # FAILED: src/ale/libale.a # /bin/sh: CMAKE_CXX_COMPILER_AR-NOTFOUND: command not found lib.systems.inspect.patterns.isDarwin ]; };}