Fix typing

Toloka · Jan 13, 2025 · 46ac2f5 · 46ac2f5
1 parent 9876ce0
commit 46ac2f5
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/crowdkit/learning/conal.py b/crowdkit/learning/conal.py
@@ -36,7 +36,7 @@ def _identity_init(shape: Union[Tuple[int, int], Tuple[int, int, int]]) -> torch
     return torch.Tensor(out)
 
 
-class CoNAL(nn.Module):
+class CoNAL(nn.Module):  # type: ignore[misc]
     """
     Common Noise Adaptation Layers (CoNAL). This method introduces two types of confusions: worker-specific and
     global. Each is parameterized by a confusion matrix. The ratio of the two confusions is determined by the

diff --git a/crowdkit/learning/crowd_layer.py b/crowdkit/learning/crowd_layer.py
@@ -85,7 +85,7 @@ def crowd_layer_vw_b(
     return scale[workers] * outputs + bias[workers]
 
 
-class CrowdLayer(nn.Module):
+class CrowdLayer(nn.Module):  # type: ignore[misc]
     """
     CrowdLayer module for classification tasks.