From 17bf056930fd0380a1edb91046a2cfe50ef7d922 Mon Sep 17 00:00:00 2001 From: generatedunixname89002005307016 Date: Fri, 2 Aug 2024 12:31:56 -0700 Subject: [PATCH] Add missing Pyre mode headers] [batch:23/25] [shard:2/N] Differential Revision: D60653035 fbshipit-source-id: 99d82e2f6868b87076979c850e0242db120c75b0 --- tests/metrics/aggregation/test_cov.py | 2 ++ torcheval/metrics/aggregation/cov.py | 2 ++ 2 files changed, 4 insertions(+) diff --git a/tests/metrics/aggregation/test_cov.py b/tests/metrics/aggregation/test_cov.py index 831f77f..00aee9b 100644 --- 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 index d1808ac..7ac3f26 100644 --- 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