admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-06-19 01:07:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

style(profiling): satisfy pre-commit checks

Browse Source
This commit is contained in:
Pepijn
2026-04-21 18:16:00 +02:00
parent fe78f8fee9
commit 929400cd44
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_model_profiling.py
+1 -1
View File
@@ -275,7 +275,7 @@ def test_deterministic_forward_artifacts_infers_image_keys_without_dataset_meta(
def forward(self, batch):
image = batch["observation.images.front"]
assert image.dtype == torch.float32
assert torch.all((0.0 <= image) & (image <= 1.0))
assert torch.all((image >= 0.0) & (image <= 1.0))
return image.sum(), {"image": image}
dataset = [{"observation.images.front": torch.tensor([[[0, 255]]], dtype=torch.uint8)}]