From 46ac2f5b11fa924eed5b3c21bc0bb2647360bae6 Mon Sep 17 00:00:00 2001 From: Dmitry Ustalov Date: Mon, 13 Jan 2025 20:57:06 +0100 Subject: [PATCH] Fix typing --- crowdkit/learning/conal.py | 2 +- crowdkit/learning/crowd_layer.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/crowdkit/learning/conal.py b/crowdkit/learning/conal.py index 0f499b32..18ffcd88 100644 --- 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 index 0ad06d9f..113c1c24 100644 --- 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.