Skip to content

Commit 01bb0a7

Browse files
committed
fix lint mypy
1 parent 4261fca commit 01bb0a7

File tree

2 files changed

+2
-4
lines changed

2 files changed

+2
-4
lines changed

src/autointent/_dump_tools/base.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@
1515
ModuleAttributes: TypeAlias = (
1616
ModuleSimpleAttributes
1717
| TagsList
18-
| np.ndarray # type: ignore[type-arg]
18+
| np.ndarray
1919
| Embedder
2020
| VectorIndex
2121
| BaseEstimator

src/autointent/modules/scoring/_gcn/gcn_model.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -72,9 +72,7 @@ def __init__(
7272
self.register_buffer("label_embeddings", torch.zeros(num_classes, label_embedding_dim))
7373

7474
@staticmethod
75-
def create_correlation_matrix(
76-
train_labels: torch.Tensor, num_classes: int, p: float, tau: float
77-
) -> torch.Tensor:
75+
def create_correlation_matrix(train_labels: torch.Tensor, num_classes: int, p: float, tau: float) -> torch.Tensor:
7876
co_occurrence = train_labels.T @ train_labels
7977
num_labels_per_class = torch.diagonal(co_occurrence)
8078

0 commit comments

Comments
 (0)