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
Branches
Tags
1 merge request!126Fix pytorch pip package name
Pipeline #61749 passed
......@@ -14,6 +14,6 @@ def test_video_to_frames():
samples = [mario.Sample(v, key=i) for i, v in enumerate(video_container)]
frame_samples = VideoToFrames().transform(samples)
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.frame_id for s in frame_samples] == [0, 1, 2, 4]
......@@ -9,4 +9,4 @@ bob.bio.video
scikit-learn
scikit-image
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