From 634465476a9059bb0f7da24abd7b9b89b3f9e7ab Mon Sep 17 00:00:00 2001 From: "Pyre Bot Jr." Date: Thu, 17 Jul 2025 12:55:25 -0700 Subject: [PATCH] Add missing Pyre mode headers] [batch:1/N] [A] [A] (#1091) Summary: X-link: https://github.com/pytorch/torchrec/pull/3204 Pull Request resolved: https://github.com/pytorch/torchx/pull/1091 X-link: https://github.com/pytorch/tnt/pull/1016 Differential Revision: D78138183 --- torchx/distributed/__init__.py | 2 ++ torchx/distributed/test/dist_test.py | 2 ++ torchx/examples/apps/compute_world_size/config/__init__.py | 2 ++ 3 files changed, 6 insertions(+) diff --git a/torchx/distributed/__init__.py b/torchx/distributed/__init__.py index 35324e137..8e34f383e 100644 --- a/torchx/distributed/__init__.py +++ b/torchx/distributed/__init__.py @@ -4,6 +4,8 @@ # This source code is licensed under the BSD-style license found in the # LICENSE file in the root directory of this source tree. +# pyre-strict + """ Convenience methods to use ``torch.distributed``. """ diff --git a/torchx/distributed/test/dist_test.py b/torchx/distributed/test/dist_test.py index 0d53515b4..ccadce695 100644 --- a/torchx/distributed/test/dist_test.py +++ b/torchx/distributed/test/dist_test.py @@ -4,6 +4,8 @@ # This source code is licensed under the BSD-style license found in the # LICENSE file in the root directory of this source tree. +# pyre-strict + import os import unittest from pathlib import Path diff --git a/torchx/examples/apps/compute_world_size/config/__init__.py b/torchx/examples/apps/compute_world_size/config/__init__.py index 2e41cd717..581f84e46 100644 --- a/torchx/examples/apps/compute_world_size/config/__init__.py +++ b/torchx/examples/apps/compute_world_size/config/__init__.py @@ -3,3 +3,5 @@ # # This source code is licensed under the BSD-style license found in the # LICENSE file in the root directory of this source tree. + +# pyre-strict