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

Differential Revision: D60653035 fbshipit-source-id: 99d82e2f6868b87076979c850e0242db120c75b0
pytorch · Aug 2, 2024 · 17bf056 · 17bf056
1 parent 05aa755
commit 17bf056
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/tests/metrics/aggregation/test_cov.py b/tests/metrics/aggregation/test_cov.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 torch
 from torcheval.metrics import Covariance
 from torcheval.utils.test_utils.metric_class_tester import MetricClassTester

diff --git a/torcheval/metrics/aggregation/cov.py b/torcheval/metrics/aggregation/cov.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 collections.abc import Iterable
 from typing import Tuple, TypeVar