scope 0.5.4__tar.gz → 0.5.5__tar.gz

{scope-0.5.4 → scope-0.5.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: scope
-Version: 0.5.4
+Version: 0.5.5
 Summary: Metrics logging and analysis
 Home-page: http://github.com/danijar/scope
 Classifier: Intended Audience :: Science/Research

{scope-0.5.4 → scope-0.5.5}/scope/__init__.py

@@ -1,4 +1,4 @@
-__version__ = '0.5.4'
+__version__ = '0.5.5'
 
 from .reader import Reader
 from .writer import Writer

{scope-0.5.4 → scope-0.5.5}/scope/formats.py

@@ -194,6 +194,8 @@ class MediapyVideo:
         x.shape[-1] in (1, 3))
 
   def convert(self, x):
+    if x.shape[-1] == 1:
+      x = x.repeat(3, -1)
     return x
 
   def create(self, path):

{scope-0.5.4 → scope-0.5.5}/tests/test_video.py

@@ -1,16 +1,22 @@
 import pathlib
 
-import scope
 import numpy as np
+import pytest
+import scope
 
 
 class TestVideo:
 
-  def test_roundtrip(self, tmpdir):
+  @pytest.mark.parametrize('channels', (3, 1))
+  @pytest.mark.parametrize('fmt', (
+      scope.formats.Video(fps=10),
+      scope.formats.MediapyVideo(fps=10),
+  ))
+  def test_roundtrip(self, tmpdir, fmt, channels):
     logdir = pathlib.Path(tmpdir)
-    writer = scope.Writer(logdir, workers=0)
-    vid1 = np.ones((5, 64, 128, 3), np.uint8) + 12
-    vid2 = np.ones((5, 64, 128, 3), np.uint8) + 255
+    writer = scope.Writer(logdir, workers=0, formats=[fmt])
+    vid1 = np.ones((5, 64, 128, channels), np.uint8) + 12
+    vid2 = np.ones((5, 64, 128, channels), np.uint8) + 255
     writer.add(0, {'foo': vid1})
     writer.add(5, {'foo': vid2})
     writer.flush()