refactor(preprocessor): add init, change signiture

gnes-ai · Aug 26, 2019 · 7493af9 · 7493af9
1 parent 293d80e
commit 7493af9
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 2 deletions.
diff --git a/gnes/preprocessor/helper.py b/gnes/preprocessor/helper.py
@@ -135,7 +135,7 @@ def split_video_frames(buffer_data: bytes,
     return [np.array(Image.open(io.BytesIO(chunk))) for chunk in chunks]
 
 
-def get_gif(images, fps=4):
+def get_gif(images: 'np.ndarray', fps=10):
     cmd = ['ffmpeg', '-y',
            '-f', 'rawvideo',
            '-vcodec', 'rawvideo',

diff --git a/gnes/preprocessor/io_utils/__init__.py b/gnes/preprocessor/io_utils/__init__.py
diff --git a/gnes/preprocessor/io_utils/gif.py b/gnes/preprocessor/io_utils/gif.py
@@ -47,7 +47,7 @@ def decode_gif(data: bytes, fps: int = -1,
 
 
 def encode_gif(
-        images: List[np.ndarray],
+        images: np.ndarray,
         scale: str,
         fps: int,
         pix_fmt: str = 'rgb24'):