python3Packages.gymnasium: add atari optional-dependencies

2025-07-12 12:10:13 +02:00
parent f005c86442
commit 3fed5dff1c
1 changed files with 10 additions and 0 deletions
@@ -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 = [