small merge

Bam4d · Oct 10, 2023 · 6e50a81 · 6e50a81
1 parent 40dad49
commit 6e50a81
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/python/griddly/util/environment_generator_generator.py b/python/griddly/util/environment_generator_generator.py
@@ -1,5 +1,5 @@
 import os
-from typing import Optional, List, Any, Dict
+from typing import Optional, List, Any, Dict, Union
 import gymnasium as gym
 import numpy as np
 import yaml

diff --git a/python/griddly/util/rllib/environment/observer_episode_recorder.py b/python/griddly/util/rllib/environment/observer_episode_recorder.py
@@ -5,8 +5,7 @@
 
 from griddly.gym import GymWrapper
 from griddly.util.render_tools import RenderToVideo
-from griddly.wrappers import RenderWrapper
-
+from griddly.wrappers.render_wrapper import RenderWrapper
 
 class RecordingState(Enum):
     NOT_RECORDING = 1