Prechádzať zdrojové kódy

Add packages to setup.py (#250)

Rodrigo de Lazcano 2 rokov pred
rodič
commit
7d16b4a60f
2 zmenil súbory, kde vykonal 2 pridanie a 2 odobranie
  1. 0 0
      minigrid/minigrid_env.py
  2. 2 2
      setup.py

+ 0 - 0
minigrid/minigrid_env.py


+ 2 - 2
setup.py

@@ -1,4 +1,4 @@
-from setuptools import setup
+from setuptools import find_packages, setup
 
 with open("README.md") as fh:
     long_description = ""
@@ -31,7 +31,7 @@ setup(
     url="https://github.com/Farama-Foundation/MiniGrid",
     description="Minimalistic gridworld reinforcement learning environments",
     extras_require=extras,
-    packages=["minigrid", "minigrid.envs"],
+    packages=[package for package in find_packages() if package.startswith("minigrid")],
     entry_points={
         "gymnasium.envs": ["__root__ = minigrid.__init__:register_minigrid_envs"]
     },