Pyre Configurationless migration for] [batch:231/244]

Reviewed By: inseokhwang

Differential Revision: D54686975

fbshipit-source-id: 58759f1c69c8abc06cf32b9a9f406dc4c51de2d5

Author: connernilsen

benchmarks/torchrec/main.py

@@ -5,6 +5,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 argparse
 import logging
 import os

examples/ddp_example.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 argparse
 import os
 import uuid

examples/simple_example.py

@@ -5,6 +5,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 argparse
 import uuid
 from typing import Optional

examples/torchrec/main.py

@@ -5,6 +5,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 argparse
 import os
 import uuid

tests/conftest.py

@@ -5,6 +5,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
+
 from typing import Generator
 
 import pytest

tests/gpu_tests/test_dtensor_io_preparer.py

@@ -5,6 +5,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
+
 from typing import List, Sequence, Set, Tuple
 
 import numpy as np

tests/gpu_tests/test_dtensor_utils.py

@@ -5,6 +5,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 torch
 
 import torch.distributed as dist

tests/gpu_tests/test_manifest_utils.py

@@ -5,6 +5,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 torch
 
 import torch.distributed as dist

tests/gpu_tests/test_partitioner_dtensor.py

@@ -5,6 +5,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 uuid
 from collections import defaultdict
 from typing import List

tests/gpu_tests/test_snapshot_dtensor.py

@@ -5,6 +5,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 logging
 import uuid
 from typing import Optional