diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml index 781d70a..555394d 100644 --- a/.pre-commit-config.yaml +++ b/.pre-commit-config.yaml @@ -8,7 +8,7 @@ repos: - id: pyupgrade - repo: https://github.com/bwhmather/ssort - rev: 0.12.2 + rev: 0.12.3 hooks: - id: ssort @@ -24,7 +24,7 @@ repos: args: [--in-place] - repo: https://github.com/psf/black - rev: 23.12.1 + rev: 24.1.1 hooks: - id: black args: [--line-length, "88"] diff --git a/src/spacy_wrap/architectures.py b/src/spacy_wrap/architectures.py index 5c84292..cbffa46 100644 --- a/src/spacy_wrap/architectures.py +++ b/src/spacy_wrap/architectures.py @@ -14,7 +14,6 @@ ClassificationTransformerModel instead of TransformerModel """ - from typing import Callable, List from spacy.tokens import Doc diff --git a/src/spacy_wrap/util.py b/src/spacy_wrap/util.py index c5b8c18..24c08a7 100644 --- a/src/spacy_wrap/util.py +++ b/src/spacy_wrap/util.py @@ -11,7 +11,6 @@ - split_by_doc. Changed to fetch logits instead of token embeddings """ - from typing import List, Literal import numpy as np diff --git a/tests/test_seq_clf_transformer.py b/tests/test_seq_clf_transformer.py index 32bbdba..4f6c148 100644 --- a/tests/test_seq_clf_transformer.py +++ b/tests/test_seq_clf_transformer.py @@ -1,4 +1,5 @@ """Test for the sequence classification transformer pipeline component.""" + import shutil import pytest diff --git a/tests/test_tok_clf_transformer.py b/tests/test_tok_clf_transformer.py index 8a06639..eccfe73 100644 --- a/tests/test_tok_clf_transformer.py +++ b/tests/test_tok_clf_transformer.py @@ -1,4 +1,5 @@ """Test for the token classification transformer pipeline component.""" + import shutil import pytest