Merge branch 'freeze-model-summary' of github.com:YChienHung/pytorch-lightning into freeze-model-summary

Author: YChienHung

.github/checkgroup.yml

@@ -71,7 +71,7 @@ subprojects:
   #    paths:
   #      # tpu CI availability is very limited, so we only require tpu tests
   #      # to pass when their configurations are modified
-  #      - ".github/workflows/tpu-tests.yml"
+  #      - ".github/workflows/tpu-tests.yml.disabled"
   #      - "tests/tests_pytorch/run_tpu_tests.sh"
   #    checks:
   #      - "test-on-tpus (pytorch, pjrt, v4-8)"
@@ -181,7 +181,7 @@ subprojects:
   #    paths:
   #      # tpu CI availability is very limited, so we only require tpu tests
   #      # to pass when their configurations are modified
-  #      - ".github/workflows/tpu-tests.yml"
+  #      - ".github/workflows/tpu-tests.yml.disabled"
   #      - "tests/tests_fabric/run_tpu_tests.sh"
   #    checks:
   #      - "test-on-tpus (pytorch, pjrt, v4-8)"

.github/workflows/tpu-tests.yml .github/workflows/tpu-tests.yml.disabled.github/workflows/tpu-tests.yml renamed to .github/workflows/tpu-tests.yml.disabled

@@ -381,7 +381,7 @@ def log(
         logger: Optional[bool] = None,
         on_step: Optional[bool] = None,
         on_epoch: Optional[bool] = None,
-        reduce_fx: Union[str, Callable] = "mean",
+        reduce_fx: Union[str, Callable[[Any], Any]] = "mean",
         enable_graph: bool = False,
         sync_dist: bool = False,
         sync_dist_group: Optional[Any] = None,
@@ -546,7 +546,7 @@ def log_dict(
         logger: Optional[bool] = None,
         on_step: Optional[bool] = None,
         on_epoch: Optional[bool] = None,
-        reduce_fx: Union[str, Callable] = "mean",
+        reduce_fx: Union[str, Callable[[Any], Any]] = "mean",
         enable_graph: bool = False,
         sync_dist: bool = False,
         sync_dist_group: Optional[Any] = None,