move post_install to balrog scripts (balrog-ai#9)

* move post_install to balrog scripts * update setup.py
richielo · Dec 3, 2024 · c905ac2 · c905ac2
1 parent b6d4a58
commit c905ac2
Show file tree

Hide file tree

Showing 4 changed files with 11 additions and 2 deletions.
diff --git a/balrog/scripts/__init__.py b/balrog/scripts/__init__.py
diff --git a/post_install.py → balrog/scripts/post_install.py b/post_install.py → balrog/scripts/post_install.py
diff --git a/docker/Dockerfile b/docker/Dockerfile
@@ -33,6 +33,6 @@ COPY . /opt/balrog/
 WORKDIR /opt/balrog
 
 RUN pip install '.'
-RUN python post_install.py
+RUN balrog-post-install
 
 CMD ["/bin/bash"]
diff --git a/setup.py b/setup.py
@@ -37,7 +37,7 @@
     ],
     entry_points={
         "console_scripts": [
-            "balrog-post-install=post_install:main",
+            "balrog-post-install=balrog.scripts.post_install:main",
         ],
     },
     extras_require={
@@ -52,6 +52,15 @@
     },
     package_dir={"": "./"},
     packages=setuptools.find_packages(where="./", include=["balrog*"]),
+    package_data={
+        "balrog": [
+            "config/config.yaml",
+            "environments/nle/achievements.json",
+            "environments/nle/Hack-Regular.ttf",
+            "environments/nle/tiles.pkl",
+            "environments/nle/Tiles16x16.png",
+        ]
+    },
     include_package_data=True,
     python_requires=">=3.8",
 )