Update pre-commit & run on all files

This commit is contained in:
Simon Alibert 2024-03-25 17:29:35 +01:00
parent 0fae5b206b
commit bcfdba109f
6 changed files with 7 additions and 3 deletions

View File

@ -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]

View File

@ -1,6 +1,7 @@
"""
Various positional encodings for the transformer.
"""
import math
import torch

View File

@ -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

View File

@ -3,6 +3,7 @@ Misc functions, including distributed helpers.
Mostly copy-paste from torchvision references.
"""
import datetime
import os
import pickle

View File

@ -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