Add missing Pyre mode headers] [batch:1/N] [A] [A] (pytorch#1091)

Summary:
X-link: pytorch/torchrec#3204

Pull Request resolved: pytorch#1091

X-link: pytorch/tnt#1016

Differential Revision: D78138183

Author: Pyre Bot Jr.

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``.
 """

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

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