✅ Fine-tune coverage

ENSTA-U2IS-AI · Jan 3, 2024 · bbbca63 · bbbca63
1 parent f1f2762
commit bbbca63
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/tests/layers/test_mask.py b/tests/layers/test_mask.py
@@ -76,7 +76,7 @@ def test_conv_two_estimators_even(self, img_input_even: torch.Tensor):
 
     def test_conv_error(self):
         with pytest.raises(ValueError):
-            _ = MaskedLinear(8, 2, num_estimators=1, scale=None)
+            MaskedConv2d(10, 2, num_estimators=2, kernel_size=1, scale=None)
 
         with pytest.raises(ValueError):
-            _ = MaskedLinear(10, 2, num_estimators=1, scale=0)
+            MaskedConv2d(10, 2, num_estimators=2, kernel_size=1, scale=0)
diff --git a/tests/transforms/test_corruptions.py b/tests/transforms/test_corruptions.py
@@ -76,6 +76,7 @@ def test_gaussian_blur(self):
         transform(inputs)
         transform = GaussianBlur(0)
         transform(inputs)
+        print(transform)
 
     def test_glass_blur(self):
         with pytest.raises(ValueError):