Apply suggestions from code review

malfet · web-flow · commit 544e7cc48c99 · 2024-12-09T14:24:31.000-08:00
diff --git a/test/export/test_unflatten_training_ir.py b/test/export/test_unflatten_training_ir.py
@@ -7,10 +7,7 @@
     import test_unflatten  # @manual=fbcode//caffe2/test:test_export-library
     import testing  # @manual=fbcode//caffe2/test:test_export-library
 
-import unittest
-
 from torch.export import export_for_training
-from torch.testing._internal.common_utils import IS_MACOS
 
 
 test_classes = {}
@@ -44,15 +41,6 @@ def make_dynamic_cls(cls):
     make_dynamic_cls(test)
 del test
 
-TrainingIRUnflattenTestUnflatten.test_unflatten_eager_training_ir = (  # noqa: F821
-    unittest.skipIf(
-        IS_MACOS, "See https://github.com/pytorch/pytorch/pull/142270 for context"
-    )(
-        TrainingIRUnflattenTestUnflatten.test_unflatten_eager_training_ir  # noqa: F821
-    )
-)
-
-
 if __name__ == "__main__":
     from torch._dynamo.test_case import run_tests
 
diff --git a/test/inductor/test_aot_inductor_custom_ops.py b/test/inductor/test_aot_inductor_custom_ops.py
@@ -255,9 +255,6 @@ def forward(self, x, y):
 
 
 class AOTInductorLoggingTest(LoggingTestCase):
-    @unittest.skipIf(
-        IS_MACOS, "See https://github.com/pytorch/pytorch/pull/142270 for context"
-    )
     @make_logging_test(dynamic=logging.DEBUG)
     def test_shape_env_reuse(self, records):
         # make sure ShapeEnv is only created once and reused afterwards
