From 0c26f11baf37fb74d0a57acfc112fd205d597b05 Mon Sep 17 00:00:00 2001 From: "Pyre Bot Jr." Date: Thu, 5 Jun 2025 07:11:08 -0700 Subject: [PATCH] Add missing Pyre mode headers] Differential Revision: D76031572 --- 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