diff --git a/tests/test_jit_model_analysis.py b/tests/test_jit_model_analysis.py index 2ba3754..2e336f5 100644 --- a/tests/test_jit_model_analysis.py +++ b/tests/test_jit_model_analysis.py @@ -60,7 +60,7 @@ def forward(self, x: torch.Tensor) -> torch.Tensor: x = x.reshape(-1, 2, 5) x = self.conv(x) x = torch.flatten(x, 1) - x = 3 * self.fc(x) + 1 # pyre-ignore[9] + x = 3 * self.fc(x) + 1 return x diff --git a/tests/test_weight_init.py b/tests/test_weight_init.py index 1092c82..0bbef81 100644 --- a/tests/test_weight_init.py +++ b/tests/test_weight_init.py @@ -68,7 +68,9 @@ def test_conv_weight_init(self) -> None: spatial_dim = k_size ** 3 # Calculate fan_in and fan_out. + # pyre-fixme[61]: `spatial_dim` may not be initialized here. fan_in = c_in_dim * spatial_dim + # pyre-fixme[61]: `spatial_dim` may not be initialized here. fan_out = c_out_dim * spatial_dim # Msra weight init check.