fixing import error

smarts/env/wrappers/gif_recorder.py

@@ -1,17 +1,17 @@
 # MIT License
-# 
+#
 # Copyright (C) 2022. Huawei Technologies Co., Ltd. All rights reserved.
-# 
+#
 # Permission is hereby granted, free of charge, to any person obtaining a copy
 # of this software and associated documentation files (the "Software"), to deal
 # in the Software without restriction, including without limitation the rights
 # to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 # copies of the Software, and to permit persons to whom the Software is
 # furnished to do so, subject to the following conditions:
-# 
+#
 # The above copyright notice and this permission notice shall be included in
 # all copies or substantial portions of the Software.
-# 
+#
 # THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 # IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 # FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT. IN NO EVENT SHALL THE
@@ -20,9 +20,16 @@
 # OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
 # THE SOFTWARE.
 import os
+import sys
+import subprocess
 
-from moviepy.editor import ImageClip
-from moviepy.editor import ImageSequenceClip
+try:
+    from moviepy.editor import ImageClip
+    from moviepy.editor import ImageSequenceClip
+except:
+    subprocess.check_call([sys.executable, "-m", "pip", "install", "moviepy"])
+    from moviepy.editor import ImageClip
+    from moviepy.editor import ImageSequenceClip
 import shutil
 import time
 from pathlib import Path
@@ -32,6 +39,7 @@ class GifRecorder:
     """
     Use images(rgb_array) to create a gif file.
     """
+
     def __init__(self, dir, env):
         timestamp_str = time.strftime("%Y%m%d-%H%M%S")
         self.dir = dir + "_" + timestamp_str

smarts/env/wrappers/recorder_wrapper.py

@@ -1,17 +1,17 @@
 # MIT License
-# 
+#
 # Copyright (C) 2022. Huawei Technologies Co., Ltd. All rights reserved.
-# 
+#
 # Permission is hereby granted, free of charge, to any person obtaining a copy
 # of this software and associated documentation files (the "Software"), to deal
 # in the Software without restriction, including without limitation the rights
 # to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 # copies of the Software, and to permit persons to whom the Software is
 # furnished to do so, subject to the following conditions:
-# 
+#
 # The above copyright notice and this permission notice shall be included in
 # all copies or substantial portions of the Software.
-# 
+#
 # THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 # IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 # FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT. IN NO EVENT SHALL THE
@@ -20,8 +20,6 @@
 # OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
 # THE SOFTWARE.
 import os
-
-from moviepy.editor import *
 import gym
 import gym.envs
 
@@ -32,6 +30,7 @@ class RecorderWrapper(gym.Wrapper):
     """
     A Wrapper that interacts the gym environment with the GifRecorder to record video step by step.
     """
+
     def __init__(self, dir, env):
 
         try:
@@ -63,7 +62,7 @@ def start_recording(self):
         if self.gif_recorder is None:
             self.gif_recorder = GifRecorder(self.dir, self.env)
         image = super().render(mode="rgb_array")
-        self.gif_recorder.capture_frame(self.next_frame_id(),image)
+        self.gif_recorder.capture_frame(self.next_frame_id(), image)
         self.recording = True
 
     def stop_recording(self):
@@ -79,7 +78,7 @@ def step(self, action):
         observations, rewards, dones, infos = super().step(action)
         if self.recording == True:
             image = super().render(mode="rgb_array")
-            self.gif_recorder.capture_frame(self.next_frame_id(),image)
+            self.gif_recorder.capture_frame(self.next_frame_id(), image)
 
         return observations, rewards, dones, infos