upgrade pyre version in fbcode/vision - batch 2

Differential Revision: D57183103 fbshipit-source-id: 7e2f42ddc6a1fa02abc27a451987d67a00264cbb
facebookresearch · May 10, 2024 · 1d61132 · 1d61132
1 parent 1856b63
commit 1d61132
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/fvcore/nn/print_model_statistics.py b/fvcore/nn/print_model_statistics.py
@@ -310,7 +310,11 @@ def _get_input_sizes(iterable: Iterable[Any]) -> List[Any]:  # pyre-ignore[2,3]
 
 
 def flop_count_str(
-    flops: FlopCountAnalysis, activations: Optional[ActivationCountAnalysis] = None
+    # pyre-fixme[11]: Annotation `FlopCountAnalysis` is not defined as a type.
+    # pyre-fixme[11]: Annotation `ActivationCountAnalysis` is not defined as a type.
+    flops: FlopCountAnalysis,
+    # pyre-fixme[11]: Annotation `ActivationCountAnalysis` is not defined as a type.
+    activations: Optional[ActivationCountAnalysis] = None,
 ) -> str:
     """
     Calculates the parameters and flops of the model with the given inputs