From 29afb2fd17fe8592acf42247d727e6e4d2209588 Mon Sep 17 00:00:00 2001
From: generatedunixname89002005307016
 <generatedunixname89002005307016@meta.com>
Date: Tue, 26 Mar 2024 22:02:22 -0700
Subject: [PATCH] upgrade pyre version in `fbcode/vision` - batch 2

Differential Revision: D55395614

fbshipit-source-id: 71677892b5d6f219f6df25b4efb51fb0f6b1441b
---
 fvcore/nn/print_model_statistics.py | 5 +++++
 tests/test_activation_count.py      | 2 ++
 tests/test_flop_count.py            | 2 ++
 tests/test_jit_model_analysis.py    | 2 ++
 tests/test_param_count.py           | 1 +
 5 files changed, 12 insertions(+)

diff --git a/fvcore/nn/print_model_statistics.py b/fvcore/nn/print_model_statistics.py
index b04ea43..e3568f3 100644
--- a/fvcore/nn/print_model_statistics.py
+++ b/fvcore/nn/print_model_statistics.py
@@ -9,8 +9,13 @@
 import torch
 from torch import nn
 
+# pyre-fixme[21]: Could not find module `fvcore.nn.activation_count`.
 from .activation_count import ActivationCountAnalysis
+
+# pyre-fixme[21]: Could not find module `fvcore.nn.flop_count`.
 from .flop_count import FlopCountAnalysis
+
+# pyre-fixme[21]: Could not find module `fvcore.nn.parameter_count`.
 from .parameter_count import parameter_count
 
 
diff --git a/tests/test_activation_count.py b/tests/test_activation_count.py
index c87a4e2..8a2da66 100644
--- a/tests/test_activation_count.py
+++ b/tests/test_activation_count.py
@@ -10,6 +10,8 @@
 
 import torch
 import torch.nn as nn
+
+# pyre-fixme[21]: Could not find module `fvcore.nn.activation_count`.
 from fvcore.nn.activation_count import activation_count, ActivationCountAnalysis
 from fvcore.nn.jit_handles import Handle
 from numpy import prod
diff --git a/tests/test_flop_count.py b/tests/test_flop_count.py
index a871379..b9dc95c 100644
--- a/tests/test_flop_count.py
+++ b/tests/test_flop_count.py
@@ -9,6 +9,8 @@
 
 import torch
 import torch.nn as nn
+
+# pyre-fixme[21]: Could not find module `fvcore.nn.flop_count`.
 from fvcore.nn.flop_count import _DEFAULT_SUPPORTED_OPS, flop_count, FlopCountAnalysis
 from fvcore.nn.jit_handles import Handle
 from torch.autograd.function import Function
diff --git a/tests/test_jit_model_analysis.py b/tests/test_jit_model_analysis.py
index e300dc9..425437c 100644
--- a/tests/test_jit_model_analysis.py
+++ b/tests/test_jit_model_analysis.py
@@ -12,6 +12,8 @@
 
 import torch
 import torch.nn as nn
+
+# pyre-fixme[21]: Could not find module `fvcore.nn.flop_count`.
 from fvcore.nn.flop_count import FlopCountAnalysis
 from fvcore.nn.jit_analysis import JitModelAnalysis
 from fvcore.nn.jit_handles import addmm_flop_jit, conv_flop_jit, Handle, linear_flop_jit
diff --git a/tests/test_param_count.py b/tests/test_param_count.py
index 64482bd..0642a53 100644
--- a/tests/test_param_count.py
+++ b/tests/test_param_count.py
@@ -5,6 +5,7 @@
 
 import unittest
 
+# pyre-fixme[21]: Could not find module `fvcore.nn.parameter_count`.
 from fvcore.nn.parameter_count import parameter_count, parameter_count_table
 from torch import nn