Skip to content
Snippets Groups Projects
Commit 17d36a69 authored by Amir MOHAMMADI's avatar Amir MOHAMMADI
Browse files

Merge branch 'fix-requirements.txt' into 'master'

Fix pytorch pip package name

See merge request !126
parents bf53c26b 98dee576
No related branches found
No related tags found
1 merge request!126Fix pytorch pip package name
Pipeline #61749 passed
...@@ -14,6 +14,6 @@ def test_video_to_frames(): ...@@ -14,6 +14,6 @@ def test_video_to_frames():
samples = [mario.Sample(v, key=i) for i, v in enumerate(video_container)] samples = [mario.Sample(v, key=i) for i, v in enumerate(video_container)]
frame_samples = VideoToFrames().transform(samples) frame_samples = VideoToFrames().transform(samples)
assert len(frame_samples) == 4 assert len(frame_samples) == 4
assert all(s.key == 0 for s in frame_samples) assert all("0_" in s.key for s in frame_samples)
assert [s.data for s in frame_samples] == [0, 1, 2, 3] assert [s.data for s in frame_samples] == [0, 1, 2, 3]
assert [s.frame_id for s in frame_samples] == [0, 1, 2, 4] assert [s.frame_id for s in frame_samples] == [0, 1, 2, 4]
...@@ -9,4 +9,4 @@ bob.bio.video ...@@ -9,4 +9,4 @@ bob.bio.video
scikit-learn scikit-learn
scikit-image scikit-image
imageio-ffmpeg imageio-ffmpeg
pytorch torch
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment