[QA] Add more pre commit checks

This commit is contained in:
Alexander Engelsberger
2021-06-16 13:12:28 +02:00
committed by Alexander Engelsberger
parent 2c908a8361
commit d0ae94f2af
5 changed files with 35 additions and 11 deletions

View File

@@ -181,7 +181,7 @@ class LabeledComponents(Components):
class ReasoningComponents(Components):
"""ReasoningComponents generate a set of components and a set of reasoning matrices.
r"""ReasoningComponents generate a set of components and a set of reasoning matrices.
Every Component has a reasoning matrix assigned.

View File

@@ -89,6 +89,6 @@ def _check_shapes(signal_int_shape, proto_int_shape):
def _int_and_mixed_shape(tensor):
shape = mixed_shape(tensor)
int_shape = tuple([i if isinstance(i, int) else None for i in shape])
int_shape = tuple(i if isinstance(i, int) else None for i in shape)
return shape, int_shape

View File

@@ -10,7 +10,6 @@ class WTAC(torch.nn.Module):
Thin wrapper over the `wtac` function.
"""
def forward(self, distances, labels):
return wtac(distances, labels)
@@ -21,7 +20,6 @@ class LTAC(torch.nn.Module):
Thin wrapper over the `wtac` function.
"""
def forward(self, probs, labels):
return wtac(-1.0 * probs, labels)
@@ -32,7 +30,6 @@ class KNNC(torch.nn.Module):
Thin wrapper over the `knnc` function.
"""
def __init__(self, k=1, **kwargs):
super().__init__(**kwargs)
self.k = k