From bcfdba109f76f81c5e350dfa19567adc148523ee Mon Sep 17 00:00:00 2001
From: Simon Alibert <alibert.sim@gmail.com>
Date: Mon, 25 Mar 2024 17:29:35 +0100
Subject: [PATCH] Update pre-commit & run on all files

---
 .pre-commit-config.yaml                                 | 4 ++--
 lerobot/__version__.py                                  | 2 +-
 lerobot/common/policies/act/position_encoding.py        | 1 +
 lerobot/common/policies/act/transformer.py              | 1 +
 lerobot/common/policies/act/utils.py                    | 1 +
 lerobot/common/policies/diffusion/model/tensor_utils.py | 1 +
 6 files changed, 7 insertions(+), 3 deletions(-)

diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
index 2b79434a4..765b678a2 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 0fc9dd115..6232b699d 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 94e862f6d..63bb48404 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 b2bd3685e..20cfc8159 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 2ce920949..0d935839b 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 0801e2946..df9a568aa 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