From 929400cd44e422a2e72c749e7f0ac99505f47c6d Mon Sep 17 00:00:00 2001 From: Pepijn Date: Tue, 21 Apr 2026 18:16:00 +0200 Subject: [PATCH] style(profiling): satisfy pre-commit checks --- tests/test_model_profiling.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_model_profiling.py b/tests/test_model_profiling.py index bb2a06dfd..6797b7453 100644 --- a/tests/test_model_profiling.py +++ b/tests/test_model_profiling.py @@ -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)}]