admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-05-20 11:09:59 +00:00
Code Issues Packages Projects Releases Wiki Activity

chore(fromat): formatting code

Browse Source
This commit is contained in:
CarolinePascal
2026-05-07 11:24:16 +02:00
parent 8cd74ea8b8
commit a3c670b987
2 changed files with 8 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/lerobot/datasets/video_utils.py
+2 -1
View File
@@ -37,11 +37,12 @@ import torchvision
from datasets.features.features import register_feature from datasets.features.features import register_feature
from PIL import Image from PIL import Image
from lerobot.utils.import_utils import get_safe_default_video_backend
from .pyav_utils import ( from .pyav_utils import (
check_video_encoder_config_pyav, check_video_encoder_config_pyav,
detect_available_encoders_pyav, detect_available_encoders_pyav,
) )
from lerobot.utils.import_utils import get_safe_default_video_backend
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
tests/datasets/test_video_encoding.py
+6 -2
View File
@@ -478,7 +478,9 @@ class TestConcatenateVideoFiles:
def test_two_clips_frame_count(self, tmp_path): def test_two_clips_frame_count(self, tmp_path):
"""Output frame count equals the sum of the two input frame counts.""" """Output frame count equals the sum of the two input frame counts."""
out = tmp_path / "out.mp4" out = tmp_path / "out.mp4"
concatenate_video_files([TEST_ARTIFACTS_DIR / "clip_6frames.mp4", TEST_ARTIFACTS_DIR / "clip_4frames.mp4"], out) concatenate_video_files(
[TEST_ARTIFACTS_DIR / "clip_6frames.mp4", TEST_ARTIFACTS_DIR / "clip_4frames.mp4"], out
)
with av.open(str(out)) as container: with av.open(str(out)) as container:
total = sum(1 for _ in container.decode(video=0)) total = sum(1 for _ in container.decode(video=0))
@@ -497,7 +499,9 @@ class TestConcatenateVideoFiles:
def test_geometry_preserved(self, tmp_path): def test_geometry_preserved(self, tmp_path):
"""Output resolution, fps, codec and pixel format must match the inputs.""" """Output resolution, fps, codec and pixel format must match the inputs."""
out = tmp_path / "out.mp4" out = tmp_path / "out.mp4"
concatenate_video_files([TEST_ARTIFACTS_DIR / "clip_4frames.mp4", TEST_ARTIFACTS_DIR / "clip_4frames.mp4"], out) concatenate_video_files(
[TEST_ARTIFACTS_DIR / "clip_4frames.mp4", TEST_ARTIFACTS_DIR / "clip_4frames.mp4"], out
)
info = get_video_info(out) info = get_video_info(out)
assert info["video.height"] == 64 assert info["video.height"] == 64