v0.3dev -- non-local installation fix (#46)

For PiPy packaging.
vikashplus · Jan 3, 2023 · b09b5a2 · b09b5a2
1 parent 508b6ce
commit b09b5a2
Show file tree

Hide file tree

Showing 4 changed files with 11 additions and 0 deletions.
diff --git a/mj_envs/envs/__init__.py b/mj_envs/envs/__init__.py
diff --git a/mj_envs/robot/__init__.py b/mj_envs/robot/__init__.py
diff --git a/mj_envs/utils/__init__.py b/mj_envs/utils/__init__.py
diff --git a/setup.py b/setup.py
@@ -16,10 +16,21 @@
 def read(fname):
     return open(os.path.join(os.path.dirname(__file__), fname)).read()
 
+def package_files(directory):
+    paths = []
+    for (path, directories, filenames) in os.walk(directory):
+        for filename in filenames:
+            paths.append(os.path.join('..', path, filename))
+    return paths
+
+extra_files = package_files('mj_envs')
+
 setup(
     name='mj_envs',
     version='0.3.0',
     packages=find_packages(),
+    package_data={"": extra_files},
+    include_package_data=True,
     description='environments simulated in MuJoCo',
     long_description=read('README.md'),
     url='https://github.com/vikashplus/mj_envs.git',