From 22f0a448fdcc19f0f9a1eb115c95f923900a0f8c Mon Sep 17 00:00:00 2001
From: John Agapiou <jagapiou@google.com>
Date: Thu, 13 Jul 2023 08:50:59 -0700
Subject: [PATCH] Include dmlab2d in setup.py

PiperOrigin-RevId: 547814530
Change-Id: Ida883e4f7d286757284fae4b069407e4642dcbbd
---
 setup.py | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/setup.py b/setup.py
index 080c75a5..cee54bfa 100644
--- a/setup.py
+++ b/setup.py
@@ -94,6 +94,9 @@ def build_assets(self):
         'Operating System :: POSIX :: Linux',
         'Operating System :: MacOS :: MacOS X',
         'Programming Language :: Python :: 3 :: Only',
+        'Programming Language :: Python :: 3.9',
+        'Programming Language :: Python :: 3.10',
+        'Programming Language :: Python :: 3.11',
         'Topic :: Scientific/Engineering :: Artificial Intelligence',
     ],
     cmdclass={'build_py': BuildPy},
@@ -109,7 +112,8 @@ def build_assets(self):
         'absl-py',
         'chex',
         'dm_env',
-        # 'dmlab2d',  # Not yet available for PIP install.
+        'dmlab2d',
+        'dm-tree',
         'immutabledict',
         'ml-collections',
         'networkx',
@@ -123,14 +127,12 @@ def build_assets(self):
     extras_require={
         # Dependencies required for rllib example.
         'rllib': [
-            'dm-tree',
             'gym',
             'ray[rllib,default]==2.0.0',
             'numpy<1.23',  # Needed by Ray because it uses `np.bool`.
         ],
         # Dependencies required for pettingzoo example.
         'pettingzoo': [
-            'dm-tree',
             'gym',
             'matplotlib',
             'pettingzoo>=1.22.3',