make recent text env changes python 2/3 compatible

rucrbser · Feb 9, 2019 · e65b61f · e65b61f
1 parent 8bcd125
commit e65b61f
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/gym/envs/tests/test_envs.py b/gym/envs/tests/test_envs.py
@@ -51,6 +51,7 @@ def test_random_rollout():
 
 
 def test_env_render_result_is_immutable():
+    from six import string_types
     environs = [
         envs.make('Taxi-v2'),
         envs.make('FrozenLake-v0'),
@@ -60,5 +61,5 @@ def test_env_render_result_is_immutable():
     for env in environs:
         env.reset()
         output = env.render(mode='ansi')
-        assert isinstance(output, str)
+        assert isinstance(output, string_types)
         env.close()
diff --git a/gym/wrappers/monitoring/video_recorder.py b/gym/wrappers/monitoring/video_recorder.py
@@ -181,8 +181,9 @@ def __init__(self, output_path, frames_per_sec):
         self.frames = []
 
     def capture_frame(self, frame):
+        from six import string_types
         string = None
-        if isinstance(frame, str):
+        if isinstance(frame, string_types):
             string = frame
         elif isinstance(frame, StringIO):
             string = frame.getvalue()