diff --git a/rle_python_interface/rle_python_interface.py b/rle_python_interface/rle_python_interface.py index c563863..05f76e6 100644 --- a/rle_python_interface/rle_python_interface.py +++ b/rle_python_interface/rle_python_interface.py @@ -7,6 +7,7 @@ import numpy as np from numpy.ctypeslib import as_ctypes import os +import sys rle_lib = cdll.LoadLibrary(os.path.join(os.path.dirname(__file__), 'librle_c.so')) @@ -123,7 +124,10 @@ def loadROM(self, rom_file, core): else: raise ValueError('core must be atari|snes|genesis|game_gear|sg1000') - rle_lib.loadROM(self.obj, rom_file, core_path) + if (sys.version_info > (3, 0)): + rle_lib.loadROM(self.obj, str.encode(rom_file), str.encode(core_path)) + else: + rle_lib.loadROM(self.obj, rom_file, core_path) def act(self, actionA, actionB = 0): return rle_lib.act(self.obj, int(actionA), int(actionB)) diff --git a/setup.py b/setup.py index 366fb01..5541882 100644 --- a/setup.py +++ b/setup.py @@ -42,7 +42,7 @@ def run(self): ds.spawn(['./copy_cores.sh']) _build.build.run(self) -version = '1.1.3' +version = '1.1.4' setup(name = 'rle_python_interface', version=version, description = 'Retro Learning Environment Python Interface based on Ben Goodrich\'s work', diff --git a/test/python_interface_test.py b/test/python_interface_test.py index fe5d524..d9aa1ea 100644 --- a/test/python_interface_test.py +++ b/test/python_interface_test.py @@ -1,6 +1,6 @@ import unittest import os -from rle_python_interface import RLEInterface +from rle_python_interface.rle_python_interface import RLEInterface import numpy as np