Remote print statement

PiperOrigin-RevId: 663689923
google-ai-edge · Aug 21, 2024 · e77aa7d · e77aa7d
1 parent a54d10d
commit e77aa7d
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 1 deletion.
diff --git a/ai_edge_torch/_convert/test/test_convert_multisig.py b/ai_edge_torch/_convert/test/test_convert_multisig.py
@@ -86,6 +86,7 @@ def test_convert_mobilenet_v2_no_default(self):
 
   def test_convert_mobilenet_v2_signature_helper(self):
     """Tests the ai_edge_torch.signature helper function works."""
+    self.skipTest("b/361146498")
     torch_module = torchvision.models.mobilenet_v2().eval()
 
     args = (torch.randn(4, 3, 224, 224),)

diff --git a/ai_edge_torch/generative/examples/gemma/gemma2.py b/ai_edge_torch/generative/examples/gemma/gemma2.py
@@ -242,7 +242,6 @@ def define_and_run_2b() -> None:
   out = model.forward(tokens, input_pos)
   out_final = out[0, 8, :]
   assert torch.allclose(gemma2_goldens, out_final, atol=1e-04)
-  print(out)
 
 
 if __name__ == "__main__":