Merge pull request #75 from cassidylaidlaw/master

Fix setup.py for visualization
HumanCompatibleAI · May 7, 2021 · 3b7a161 · 3b7a161
2 parents 05aec15 + cc7ddd6
commit 3b7a161
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -18,7 +18,10 @@
       keywords=['Overcooked', 'AI', 'Reinforcement Learning'],
       package_dir={"": "src"},
       package_data={
-        'overcooked_ai_py' : ['data/layouts/*.layout', 'data/planners/*.py', 'data/human_data/*.pickle']
+        'overcooked_ai_py' : [
+          'data/layouts/*.layout', 'data/planners/*.py', 'data/human_data/*.pickle',
+          'data/graphics/*.png', 'data/graphics/*.json', 'data/fonts/*.ttf',
+        ],
       },
       install_requires=[
         'numpy',

diff --git a/src/overcooked_ai_py/visualization/__init__.py b/src/overcooked_ai_py/visualization/__init__.py