Add missing Pyre mode headers] [shard:2/N]

Differential Revision: D65751031 fbshipit-source-id: aaa13c1844544a9978c965d80b4e6eb773739d1d
pytorch · Nov 12, 2024 · 4f91877 · 4f91877
1 parent b38884b
commit 4f91877
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/torchtnt/utils/event.py b/torchtnt/utils/event.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 dataclasses import dataclass, field
 from typing import Dict, Union
 

diff --git a/torchtnt/utils/event_handlers.py b/torchtnt/utils/event_handlers.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 random
 from contextlib import contextmanager