Add packages to setup.py (#250)

Farama-Foundation · Sep 28, 2022 · 7d16b4a · 7d16b4a
1 parent 56f6a57
commit 7d16b4a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/minigrid/minigrid_env.py b/minigrid/minigrid_env.py
diff --git a/setup.py b/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 @@
     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"]
     },