forked from Farama-Foundation/Arcade-Learning-Environment
-
Notifications
You must be signed in to change notification settings - Fork 0
/
setup.py
25 lines (23 loc) · 1.05 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
from distutils.core import setup, Extension
import os.path, sys
ale_c_lib = 'ale_python_interface/libale_c.so'
if not os.path.isfile(ale_c_lib):
print('ERROR: Unable to find required library: %s. Please ensure you\'ve '\
'built ALE using CMake.'%(ale_c_lib))
sys.exit()
module1 = Extension('ale_python_interface.ale_c_wrapper',
libraries = ['ale_c'],
include_dirs = ['src'],
library_dirs = ['ale_python_interface'],
extra_compile_args=['-D__STDC_CONSTANT_MACROS'],
sources=['ale_python_interface/ale_c_wrapper.cpp'])
setup(name = 'ale_python_interface',
version='0.0.1',
description = 'Arcade Learning Environment Python Interface',
url='https://github.com/bbitmaster/ale_python_interface',
author='Ben Goodrich',
license='GPL',
ext_modules = [module1],
packages=['ale_python_interface'],
package_dir={'ale_python_interface': 'ale_python_interface'},
package_data={'ale_python_interface': ['libale_c.so']})