Add tests

phamquiluan · Aug 5, 2020 · e4a86a1 · e4a86a1
1 parent f7226da
commit e4a86a1
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tests/test_sanitize.py b/tests/test_sanitize.py
@@ -4,7 +4,9 @@
 
 
 def test_sanitize():
-    m = resattnet56(in_channels=3, num_classes=10, pretrained=False)  # pretrained is load automatically
+    m = resattnet56(
+        in_channels=3, num_classes=10, pretrained=False
+    )  # pretrained is load automatically
 
     tensor = torch.Tensor(1, 3, 224, 224)