fix oss pyre (pytorch#2640)

Summary: Pull Request resolved: pytorch#2640 tsia Reviewed By: sarckk Differential Revision: D67209515 fbshipit-source-id: 05158068db8ce32a94ee0be2bc4c3ece8f39bac8
sarckk · Dec 18, 2024 · a43cef8 · a43cef8
1 parent e6e4f6c
commit a43cef8
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/torchrec/distributed/train_pipeline/tests/test_train_pipelines.py b/torchrec/distributed/train_pipeline/tests/test_train_pipelines.py
@@ -2257,6 +2257,7 @@ def setUp(self) -> None:
         torch.autograd.set_detect_anomaly(False)
         self._exit_stack = ExitStack()
         self._exit_stack.enter_context(
+            # type: ignore[attr-defined]
             torch._dynamo.config.patch(
                 optimize_ddp="python_reducer_without_compiled_forward"
             ),