diff --git a/setup.py b/setup.py index 94a148f5..aea26bd1 100644 --- a/setup.py +++ b/setup.py @@ -3,20 +3,41 @@ # This source code is licensed under the license found in the # LICENSE file in the root directory of this source tree. # +import os +from pathlib import Path from setuptools import find_packages, setup + +def package_files(directory): + paths = [] + for path, _, filenames in os.walk(directory): + for filename in filenames: + paths.append(os.path.join("..", path, filename)) + return paths + + +extra_files = package_files( + str( + Path(os.path.dirname(os.path.realpath(__file__))) + / Path("benchmarl") + / Path("conf") + ) +) + setup( name="benchmarl", - version="0.0.2", + version="0.0.3", description="BenchMARL", url="https://github.com/facebookresearch/BenchMARL", author="Matteo Bettini", author_email="mb2389@cl.cam.ac.uk", - packages=find_packages(), install_requires=["torchrl>=0.2.0", "tqdm", "hydra-core"], extras_require={ "vmas": ["vmas>=1.2.10"], "pettingzoo": ["pettingzoo[all]>=1.24.1"], }, + packages=find_packages(), + include_package_data=True, + package_data={"": extra_files}, )