Update test_testing.py

pytorch · Nov 5, 2022 · 0a28c08 · 0a28c08
1 parent bb96453
commit 0a28c08
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/test/test_testing.py b/test/test_testing.py
@@ -1783,9 +1783,11 @@ def test_circular_dependencies(self) -> None:
         if not sys.version_info >= (3, 9):
             ignored_modules.append("torch.utils.benchmark")
         if IS_WINDOWS or IS_MACOS:
-            # Distributed should be importable on Windows but not on Mac
+            # Distributed should be importable on Windows(except nn.api.), but not on Mac
             if IS_MACOS:
                 ignored_modules.append("torch.distributed.")
+            else:
+                ignored_modules.append("torch.distributed.nn.api.")
             ignored_modules.append("torch.testing._internal.dist_utils")
             # And these both end up with transitive dependencies on distributed
             ignored_modules.append("torch.nn.parallel._replicated_tensor_ddp_interop")