diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml index 2b79434a..765b678a 100644 --- a/.pre-commit-config.yaml +++ b/.pre-commit-config.yaml @@ -14,11 +14,11 @@ repos: - id: end-of-file-fixer - id: trailing-whitespace - repo: https://github.com/asottile/pyupgrade - rev: v3.15.1 + rev: v3.15.2 hooks: - id: pyupgrade - repo: https://github.com/astral-sh/ruff-pre-commit - rev: v0.2.2 + rev: v0.3.4 hooks: - id: ruff args: [--fix] diff --git a/lerobot/__version__.py b/lerobot/__version__.py index 0fc9dd11..6232b699 100644 --- a/lerobot/__version__.py +++ b/lerobot/__version__.py @@ -1,4 +1,4 @@ -""" To enable `lerobot.__version__` """ +"""To enable `lerobot.__version__`""" from importlib.metadata import PackageNotFoundError, version diff --git a/lerobot/common/policies/act/position_encoding.py b/lerobot/common/policies/act/position_encoding.py index 94e862f6..63bb4840 100644 --- a/lerobot/common/policies/act/position_encoding.py +++ b/lerobot/common/policies/act/position_encoding.py @@ -1,6 +1,7 @@ """ Various positional encodings for the transformer. """ + import math import torch diff --git a/lerobot/common/policies/act/transformer.py b/lerobot/common/policies/act/transformer.py index b2bd3685..20cfc815 100644 --- a/lerobot/common/policies/act/transformer.py +++ b/lerobot/common/policies/act/transformer.py @@ -6,6 +6,7 @@ Copy-paste from torch.nn.Transformer with modifications: * extra LN at the end of encoder is removed * decoder returns a stack of activations from all decoding layers """ + import copy from typing import Optional diff --git a/lerobot/common/policies/act/utils.py b/lerobot/common/policies/act/utils.py index 2ce92094..0d935839 100644 --- a/lerobot/common/policies/act/utils.py +++ b/lerobot/common/policies/act/utils.py @@ -3,6 +3,7 @@ Misc functions, including distributed helpers. Mostly copy-paste from torchvision references. """ + import datetime import os import pickle diff --git a/lerobot/common/policies/diffusion/model/tensor_utils.py b/lerobot/common/policies/diffusion/model/tensor_utils.py index 0801e294..df9a568a 100644 --- a/lerobot/common/policies/diffusion/model/tensor_utils.py +++ b/lerobot/common/policies/diffusion/model/tensor_utils.py @@ -2,6 +2,7 @@ A collection of utilities for working with nested tensor structures consisting of numpy arrays and torch tensors. """ + import collections import numpy as np