Commit 3fed5dff authored by Gaetan Lepage's avatar Gaetan Lepage
Browse files

python3Packages.gymnasium: add atari optional-dependencies

parent f005c864
Loading
Loading
Loading
Loading
+10 −0
Original line number Diff line number Diff line
@@ -15,6 +15,10 @@
  pythonOlder,
  importlib-metadata,

  # optional-dependencies
  # atari
  ale-py,

  # tests
  array-api-compat,
  dill,
@@ -54,6 +58,12 @@ buildPythonPackage rec {
    typing-extensions
  ] ++ lib.optionals (pythonOlder "3.10") [ importlib-metadata ];

  optional-dependencies = {
    atari = [
      ale-py
    ];
  };

  pythonImportsCheck = [ "gymnasium" ];

  nativeCheckInputs = [