admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-07-24 10:16:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

test(fix): fixing failing tests

Browse Source
This commit is contained in:
CarolinePascal
2026-06-24 22:47:20 +02:00
parent 24c0bf6199
commit 95f2923b56
2 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/annotations/test_frames.py
+1
View File
@@ -47,6 +47,7 @@ class _FakeMeta:
def __init__(self, video_keys: list[str], image_keys: list[str], video_path: Path | None = None) -> None: def __init__(self, video_keys: list[str], image_keys: list[str], video_path: Path | None = None) -> None:
self.video_keys = video_keys self.video_keys = video_keys
self.camera_keys = [*video_keys, *image_keys] self.camera_keys = [*video_keys, *image_keys]
self.depth_keys = []
self._video_path = video_path self._video_path = video_path
self.episodes = {0: {f"videos/{key}/from_timestamp": 0.0 for key in video_keys}} self.episodes = {0: {f"videos/{key}/from_timestamp": 0.0 for key in video_keys}}
tests/datasets/test_image_writer.py
+1 -1
View File
@@ -344,7 +344,7 @@ def test_with_different_image_formats(tmp_path, img_array_factory):
writer = AsyncImageWriter() writer = AsyncImageWriter()
try: try:
image_array = img_array_factory() image_array = img_array_factory()
formats = ["png", "jpeg", "bmp"] formats = ["png", "tiff", "tif"]
for fmt in formats: for fmt in formats:
fpath = tmp_path / f"test_image.{fmt}" fpath = tmp_path / f"test_image.{fmt}"
write_image(image_array, fpath) write_image(image_array, fpath)