fix oss pyre (pytorch#2640)

Summary: tsia Reviewed By: sarckk Differential Revision: D67209515
iamzainhuda · Dec 17, 2024 · 2123bb1 · 2123bb1
1 parent e6e4f6c
commit 2123bb1
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"
             ),