newline

tests/test_model_components/test_mc_image.py

@@ -111,4 +111,4 @@ def test_all_frozen():
 def test_pretrained_model_setup_defaults(arch, expected_out_shape):
     model = Vision(pretrained_model_setup=arch, n_trainable=0)
     out = model(X_images)
-    assert out.size(0) == 10 and out.size(1) == expected_out_shape
+    assert out.size(0) == 10 and out.size(1) == expected_out_shape
\ No newline at end of file