update tests wrt odml-torch as default changes

PiperOrigin-RevId: 706043950
google-ai-edge · Dec 14, 2024 · 6c94a00 · 6c94a00
1 parent ae137e7
commit 6c94a00
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 290 deletions.
diff --git a/ai_edge_torch/hlfb/test/test_stablehlo_composite_builder.py b/ai_edge_torch/hlfb/test/test_stablehlo_composite_builder.py
diff --git a/test/test_quantize.py b/test/test_quantize.py
@@ -37,10 +37,6 @@ def setUp(self):
     super().setUp()
     torch.manual_seed(0)
 
-  @googletest.skipIf(
-      not ai_edge_torch.config.use_torch_xla,
-      reason="Only working with torch_xla at the moment.",
-  )
   def test_quantizer_arg(self):
     """Compare the sizes of models.