Include dmlab2d in setup.py

PiperOrigin-RevId: 547814530
Change-Id: Ida883e4f7d286757284fae4b069407e4642dcbbd

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',