Speed up CI, add more checks (#115)

- Split pytest and end-to-end tests into separate jobs
- Add poetry check to ensure pyproject.toml and poetry.lock are in sync
- Add ruff format --diff to ensure style formatting is applied (fails if ruff would reformat anything)
This commit is contained in:
Simon Alibert
2024-04-29 23:05:55 +02:00
committed by GitHub
parent 1ec5f77f7c
commit 2765877f28
9 changed files with 88 additions and 21 deletions

View File

@@ -14,8 +14,8 @@ from lerobot.common.datasets.utils import (
class PushTProcessor:
""" Process zarr files formatted like in: https://github.com/real-stanford/diffusion_policy
"""
"""Process zarr files formatted like in: https://github.com/real-stanford/diffusion_policy"""
def __init__(self, folder_path: Path, fps: int | None = None):
self.zarr_path = folder_path
if fps is None:

View File

@@ -195,7 +195,6 @@ class ActionChunkingTransformerPolicy(nn.Module):
return loss_dict
def _stack_images(self, batch: dict[str, Tensor]) -> dict[str, Tensor]:
"""Stacks all the images in a batch and puts them in a new key: "observation.images".