-
Notifications
You must be signed in to change notification settings - Fork 0
/
setup.py
47 lines (43 loc) · 1.58 KB
/
setup.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
from setuptools import find_packages
from distutils.core import setup, Extension
from pybind11.setup_helpers import Pybind11Extension, build_ext
import os
def getListOfFiles(dirName):
# create a list of file and sub directories
# names in the given directory
listOfFile = os.listdir(dirName)
allFiles = list()
# Iterate over all the entries
for entry in listOfFile:
# Create full path
fullPath = os.path.join(dirName, entry)
# If entry is a directory then get the list of files in this directory
if os.path.isdir(fullPath):
allFiles = allFiles + getListOfFiles(fullPath)
else:
allFiles.append(fullPath)
return allFiles
module1 = Pybind11Extension('flocking_cpp',
#sources = s)
sources = ['fle/cpp/py_interface.cpp'],
extra_compile_args = [])
setup(
name='fle',
version=0.1,
author='Caroline Horsch, J K Terry, Ben Black, Kyle Sang',
description="Flocking Learning Environment",
url="http://github.com/jkterry1/birdflocking",
keywords=["Reinforcement Learning", "game", "RL", "AI", "gym"],
options={"bdist_wheel": {"universal": True}},
packages=["fle"] + ["fle." + pkg for pkg in find_packages("fle")],
include_package_data=True,
install_requires=[
"pettingzoo>=1.11.1",
"pybind11",
"matplotlib>=3.3"
],
setup_requires=['pybind11>=2.2'],
cmdclass={"build_ext": build_ext},
ext_modules = [module1]
)