From 3fed5dff1c586196667ab92cbebd06ed4fe0d754 Mon Sep 17 00:00:00 2001
From: Gaetan Lepage <gaetan@glepage.com>
Date: Thu, 10 Jul 2025 22:37:56 +0200
Subject: [PATCH] python3Packages.gymnasium: add atari optional-dependencies

---
 pkgs/development/python-modules/gymnasium/default.nix | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/pkgs/development/python-modules/gymnasium/default.nix b/pkgs/development/python-modules/gymnasium/default.nix
index d9088c43d27c..f81312e97e58 100644
--- a/pkgs/development/python-modules/gymnasium/default.nix
+++ b/pkgs/development/python-modules/gymnasium/default.nix
@@ -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 = [