Pyre Configurationless migration for] [batch:23/25] (#841)

Summary: Pull Request resolved: #841

Reviewed By: connernilsen

Differential Revision: D57836213

fbshipit-source-id: 39b0f90fe467850e7f0a6d45a2213cd7d4186dba

Author: generatedunixname89002005287564

tests/framework/callbacks/test_early_stopping.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 unittest
 from typing import cast, Literal
 from unittest.mock import MagicMock, patch

tests/framework/callbacks/test_time_wait_for_batch_logger.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 unittest
 from unittest.mock import ANY, call, MagicMock
 

torchtnt/framework/callbacks/early_stopping.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
+
 from typing import Literal
 
 from torchtnt.framework.callback import Callback

torchtnt/framework/callbacks/progress_reporter.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 logging
 from typing import cast

torchtnt/framework/callbacks/slow_rank_detector.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 logging
 import time
 from typing import List, Optional, Tuple

torchtnt/framework/callbacks/time_wait_for_batch_logger.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
+
 
 from typing import cast, Union
 

torchtnt/utils/fsdp_utils.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 importlib
 
 from dataclasses import dataclass