pytorch-ignite 0.6.0.dev20251103__py3-none-any.whl → 0.6.0.dev20260102__py3-none-any.whl

ignite/__init__.py

@@ -6,4 +6,4 @@ import ignite.handlers
 import ignite.metrics
 import ignite.utils
 
-__version__ = "0.6.0.dev20251103"
+__version__ = "0.6.0.dev20260102"

ignite/contrib/engines/common.py

@@ -265,6 +265,7 @@ def _setup_common_distrib_training_handlers(
 
         @trainer.on(Events.EPOCH_STARTED)
         def distrib_set_epoch(engine: Engine) -> None:
+            # pyrefly: ignore [missing-attribute]
             train_sampler.set_epoch(engine.state.epoch - 1)
 
 

ignite/contrib/handlers/base_logger.py

@@ -9,7 +9,7 @@ import warnings
 removed_in = "0.6.0"
 deprecation_warning = (
     f"{__file__} has been moved to /ignite/handlers/base_logger.py"
-    + (f" and will be removed in version {removed_in}" if removed_in else "")
+    + f" and will be removed in version {removed_in}"
     + ".\n Please refer to the documentation for more details."
 )
 warnings.warn(deprecation_warning, DeprecationWarning, stacklevel=2)

ignite/contrib/handlers/clearml_logger.py

@@ -9,7 +9,7 @@ import warnings
 removed_in = "0.6.0"
 deprecation_warning = (
     f"{__file__} has been moved to /ignite/handlers/clearml_logger.py"
-    + (f" and will be removed in version {removed_in}" if removed_in else "")
+    + f" and will be removed in version {removed_in}"
     + ".\n Please refer to the documentation for more details."
 )
 warnings.warn(deprecation_warning, DeprecationWarning, stacklevel=2)

ignite/contrib/handlers/lr_finder.py

@@ -9,7 +9,7 @@ import warnings
 removed_in = "0.6.0"
 deprecation_warning = (
     f"{__file__} has been moved to /ignite/handlers/lr_finder.py"
-    + (f" and will be removed in version {removed_in}" if removed_in else "")
+    + f" and will be removed in version {removed_in}"
     + ".\n Please refer to the documentation for more details."
 )
 warnings.warn(deprecation_warning, DeprecationWarning, stacklevel=2)

ignite/contrib/handlers/mlflow_logger.py

@@ -9,7 +9,7 @@ import warnings
 removed_in = "0.6.0"
 deprecation_warning = (
     f"{__file__} has been moved to /ignite/handlers/mlflow_logger.py"
-    + (f" and will be removed in version {removed_in}" if removed_in else "")
+    + f" and will be removed in version {removed_in}"
     + ".\n Please refer to the documentation for more details."
 )
 warnings.warn(deprecation_warning, DeprecationWarning, stacklevel=2)

ignite/contrib/handlers/neptune_logger.py

@@ -9,7 +9,7 @@ import warnings
 removed_in = "0.6.0"
 deprecation_warning = (
     f"{__file__} has been moved to /ignite/handlers/neptune_logger.py"
-    + (f" and will be removed in version {removed_in}" if removed_in else "")
+    + f" and will be removed in version {removed_in}"
     + ".\n Please refer to the documentation for more details."
 )
 warnings.warn(deprecation_warning, DeprecationWarning, stacklevel=2)

ignite/contrib/handlers/param_scheduler.py

@@ -9,7 +9,7 @@ import warnings
 removed_in = "0.6.0"
 deprecation_warning = (
     f"{__file__} has been moved to /ignite/handlers/param_scheduler.py"
-    + (f" and will be removed in version {removed_in}" if removed_in else "")
+    + f" and will be removed in version {removed_in}"
     + ".\n Please refer to the documentation for more details."
 )
 warnings.warn(deprecation_warning, DeprecationWarning, stacklevel=2)

ignite/contrib/handlers/polyaxon_logger.py

@@ -9,7 +9,7 @@ import warnings
 removed_in = "0.6.0"
 deprecation_warning = (
     f"{__file__} has been moved to /ignite/handlers/polyaxon_logger.py"
-    + (f" and will be removed in version {removed_in}" if removed_in else "")
+    + f" and will be removed in version {removed_in}"
     + ".\n Please refer to the documentation for more details."
 )
 warnings.warn(deprecation_warning, DeprecationWarning, stacklevel=2)

ignite/contrib/handlers/tensorboard_logger.py

@@ -9,7 +9,7 @@ import warnings
 removed_in = "0.6.0"
 deprecation_warning = (
     f"{__file__} has been moved to /ignite/handlers/tensorboard_logger.py"
-    + (f" and will be removed in version {removed_in}" if removed_in else "")
+    + f" and will be removed in version {removed_in}"
     + ".\n Please refer to the documentation for more details."
 )
 warnings.warn(deprecation_warning, DeprecationWarning, stacklevel=2)

ignite/contrib/handlers/time_profilers.py

@@ -9,7 +9,7 @@ import warnings
 removed_in = "0.6.0"
 deprecation_warning = (
     f"{__file__} has been moved to /ignite/handlers/time_profilers.py"
-    + (f" and will be removed in version {removed_in}" if removed_in else "")
+    + f" and will be removed in version {removed_in}"
     + ".\n Please refer to the documentation for more details."
 )
 warnings.warn(deprecation_warning, DeprecationWarning, stacklevel=2)

ignite/contrib/handlers/tqdm_logger.py

@@ -9,7 +9,7 @@ import warnings
 removed_in = "0.6.0"
 deprecation_warning = (
     f"{__file__} has been moved to /ignite/handlers/tqdm_logger.py"
-    + (f" and will be removed in version {removed_in}" if removed_in else "")
+    + f" and will be removed in version {removed_in}"
     + ".\n Please refer to the documentation for more details."
 )
 warnings.warn(deprecation_warning, DeprecationWarning, stacklevel=2)

ignite/contrib/handlers/visdom_logger.py

@@ -9,7 +9,7 @@ import warnings
 removed_in = "0.6.0"
 deprecation_warning = (
     f"{__file__} has been moved to /ignite/handlers/visdom_logger.py"
-    + (f" and will be removed in version {removed_in}" if removed_in else "")
+    + f" and will be removed in version {removed_in}"
     + ".\n Please refer to the documentation for more details."
 )
 warnings.warn(deprecation_warning, DeprecationWarning, stacklevel=2)

ignite/contrib/handlers/wandb_logger.py

@@ -9,7 +9,7 @@ import warnings
 removed_in = "0.6.0"
 deprecation_warning = (
     f"{__file__} has been moved to /ignite/handlers/wandb_logger.py"
-    + (f" and will be removed in version {removed_in}" if removed_in else "")
+    + f" and will be removed in version {removed_in}"
     + ".\n Please refer to the documentation for more details."
 )
 warnings.warn(deprecation_warning, DeprecationWarning, stacklevel=2)

ignite/contrib/metrics/average_precision.py

@@ -9,7 +9,7 @@ import warnings
 removed_in = "0.6.0"
 deprecation_warning = (
     f"{__file__} has been moved to /ignite/metrics/average_precision.py"
-    + (f" and will be removed in version {removed_in}" if removed_in else "")
+    + f" and will be removed in version {removed_in}"
     + ".\n Please refer to the documentation for more details."
 )
 warnings.warn(deprecation_warning, DeprecationWarning, stacklevel=2)

ignite/contrib/metrics/cohen_kappa.py

@@ -9,7 +9,7 @@ import warnings
 removed_in = "0.6.0"
 deprecation_warning = (
     f"{__file__} has been moved to ignite/metrics/cohen_kappa.py"
-    + (f" and will be removed in version {removed_in}" if removed_in else "")
+    + f" and will be removed in version {removed_in}"
     + ".\n Please refer to the documentation for more details."
 )
 warnings.warn(deprecation_warning, DeprecationWarning, stacklevel=2)

ignite/contrib/metrics/gpu_info.py

@@ -9,7 +9,7 @@ import warnings
 removed_in = "0.6.0"
 deprecation_warning = (
     f"{__file__} has been moved to ignite/metrics/gpu_info.py"
-    + (f" and will be removed in version {removed_in}" if removed_in else "")
+    + f" and will be removed in version {removed_in}"
     + ".\n Please refer to the documentation for more details."
 )
 warnings.warn(deprecation_warning, DeprecationWarning, stacklevel=2)

ignite/contrib/metrics/precision_recall_curve.py

@@ -9,7 +9,7 @@ import warnings
 removed_in = "0.6.0"
 deprecation_warning = (
     f"{__file__} has been moved to ignite/metrics/precision_recall_curve.py"
-    + (f" and will be removed in version {removed_in}" if removed_in else "")
+    + f" and will be removed in version {removed_in}"
     + ".\n Please refer to the documentation for more details."
 )
 warnings.warn(deprecation_warning, DeprecationWarning, stacklevel=2)

ignite/contrib/metrics/regression/canberra_metric.py

@@ -9,9 +9,8 @@ import warnings
 removed_in = "0.6.0"
 deprecation_warning = (
     f"{__file__} has been moved to ignite/metrics/regression/canberra_metric.py"
-    f" and will be removed in version {removed_in}"
-    if removed_in
-    else "" ".\n Please refer to the documentation for more details."
+    + f" and will be removed in version {removed_in}"
+    + ".\n Please refer to the documentation for more details."
 )
 warnings.warn(deprecation_warning, DeprecationWarning, stacklevel=2)
 from ignite.metrics.regression.canberra_metric import CanberraMetric

ignite/contrib/metrics/regression/fractional_absolute_error.py

@@ -9,9 +9,8 @@ import warnings
 removed_in = "0.6.0"
 deprecation_warning = (
     f"{__file__} has been moved to ignite/metrics/regression/fractional_absolute_error.py"
-    f" and will be removed in version {removed_in}"
-    if removed_in
-    else "" ".\n Please refer to the documentation for more details."
+    + f" and will be removed in version {removed_in}"
+    + ".\n Please refer to the documentation for more details."
 )
 warnings.warn(deprecation_warning, DeprecationWarning, stacklevel=2)
 from ignite.metrics.regression.fractional_absolute_error import FractionalAbsoluteError

ignite/contrib/metrics/regression/fractional_bias.py

@@ -9,9 +9,8 @@ import warnings
 removed_in = "0.6.0"
 deprecation_warning = (
     f"{__file__} has been moved to ignite/metrics/regression/fractional_bias.py"
-    f" and will be removed in version {removed_in}"
-    if removed_in
-    else "" ".\n Please refer to the documentation for more details."
+    + f" and will be removed in version {removed_in}"
+    + ".\n Please refer to the documentation for more details."
 )
 warnings.warn(deprecation_warning, DeprecationWarning, stacklevel=2)
 from ignite.metrics.regression.fractional_bias import FractionalBias

ignite/contrib/metrics/regression/geometric_mean_absolute_error.py

@@ -9,9 +9,8 @@ import warnings
 removed_in = "0.6.0"
 deprecation_warning = (
     f"{__file__} has been moved to ignite/metrics/regression/geometric_mean_absolute_error.py"
-    f" and will be removed in version {removed_in}"
-    if removed_in
-    else "" ".\n Please refer to the documentation for more details."
+    + f" and will be removed in version {removed_in}"
+    + ".\n Please refer to the documentation for more details."
 )
 warnings.warn(deprecation_warning, DeprecationWarning, stacklevel=2)
 from ignite.metrics.regression.geometric_mean_absolute_error import GeometricMeanAbsoluteError

ignite/contrib/metrics/regression/geometric_mean_relative_absolute_error.py

@@ -9,9 +9,8 @@ import warnings
 removed_in = "0.6.0"
 deprecation_warning = (
     f"{__file__} has been moved to ignite/metrics/regression/geometric_mean_relative_absolute_error.py"
-    f" and will be removed in version {removed_in}"
-    if removed_in
-    else "" ".\n Please refer to the documentation for more details."
+    + f" and will be removed in version {removed_in}"
+    + ".\n Please refer to the documentation for more details."
 )
 warnings.warn(deprecation_warning, DeprecationWarning, stacklevel=2)
 from ignite.metrics.regression.geometric_mean_relative_absolute_error import GeometricMeanRelativeAbsoluteError

ignite/contrib/metrics/regression/manhattan_distance.py

@@ -9,9 +9,8 @@ import warnings
 removed_in = "0.6.0"
 deprecation_warning = (
     f"{__file__} has been moved to ignite/metrics/regression/manhattan_distance.py"
-    f" and will be removed in version {removed_in}"
-    if removed_in
-    else "" ".\n Please refer to the documentation for more details."
+    + f" and will be removed in version {removed_in}"
+    + ".\n Please refer to the documentation for more details."
 )
 warnings.warn(deprecation_warning, DeprecationWarning, stacklevel=2)
 from ignite.metrics.regression.manhattan_distance import ManhattanDistance

ignite/contrib/metrics/regression/maximum_absolute_error.py

@@ -9,9 +9,8 @@ import warnings
 removed_in = "0.6.0"
 deprecation_warning = (
     f"{__file__} has been moved to ignite/metrics/regression/maximum_absolute_error.py"
-    f" and will be removed in version {removed_in}"
-    if removed_in
-    else "" ".\n Please refer to the documentation for more details."
+    + f" and will be removed in version {removed_in}"
+    + ".\n Please refer to the documentation for more details."
 )
 warnings.warn(deprecation_warning, DeprecationWarning, stacklevel=2)
 from ignite.metrics.regression.maximum_absolute_error import MaximumAbsoluteError

ignite/contrib/metrics/regression/mean_absolute_relative_error.py

@@ -9,9 +9,8 @@ import warnings
 removed_in = "0.6.0"
 deprecation_warning = (
     f"{__file__} has been moved to ignite/metrics/regression/mean_absolute_relative_error.py"
-    f" and will be removed in version {removed_in}"
-    if removed_in
-    else "" ".\n Please refer to the documentation for more details."
+    + f" and will be removed in version {removed_in}"
+    + ".\n Please refer to the documentation for more details."
 )
 warnings.warn(deprecation_warning, DeprecationWarning, stacklevel=2)
 from ignite.metrics.regression.mean_absolute_relative_error import MeanAbsoluteRelativeError

ignite/contrib/metrics/regression/mean_error.py

@@ -9,9 +9,8 @@ import warnings
 removed_in = "0.6.0"
 deprecation_warning = (
     f"{__file__} has been moved to ignite/metrics/regression/mean_error.py"
-    f" and will be removed in version {removed_in}"
-    if removed_in
-    else "" ".\n Please refer to the documentation for more details."
+    + f" and will be removed in version {removed_in}"
+    + ".\n Please refer to the documentation for more details."
 )
 warnings.warn(deprecation_warning, DeprecationWarning, stacklevel=2)
 from ignite.metrics.regression.mean_error import MeanError

ignite/contrib/metrics/regression/mean_normalized_bias.py

@@ -9,9 +9,8 @@ import warnings
 removed_in = "0.6.0"
 deprecation_warning = (
     f"{__file__} has been moved to ignite/metrics/regression/mean_normalized_bias.py"
-    f" and will be removed in version {removed_in}"
-    if removed_in
-    else "" ".\n Please refer to the documentation for more details."
+    + f" and will be removed in version {removed_in}"
+    + ".\n Please refer to the documentation for more details."
 )
 warnings.warn(deprecation_warning, DeprecationWarning, stacklevel=2)
 from ignite.metrics.regression.mean_normalized_bias import MeanNormalizedBias

ignite/contrib/metrics/regression/median_absolute_error.py

@@ -9,9 +9,8 @@ import warnings
 removed_in = "0.6.0"
 deprecation_warning = (
     f"{__file__} has been moved to ignite/metrics/regression/median_absolute_error.py"
-    f" and will be removed in version {removed_in}"
-    if removed_in
-    else "" ".\n Please refer to the documentation for more details."
+    + f" and will be removed in version {removed_in}"
+    + ".\n Please refer to the documentation for more details."
 )
 warnings.warn(deprecation_warning, DeprecationWarning, stacklevel=2)
 from ignite.metrics.regression.median_absolute_error import MedianAbsoluteError

ignite/contrib/metrics/regression/median_absolute_percentage_error.py

@@ -9,9 +9,8 @@ import warnings
 removed_in = "0.6.0"
 deprecation_warning = (
     f"{__file__} has been moved to ignite/metrics/regression/median_absolute_percentage_error.py"
-    f" and will be removed in version {removed_in}"
-    if removed_in
-    else "" ".\n Please refer to the documentation for more details."
+    + f" and will be removed in version {removed_in}"
+    + ".\n Please refer to the documentation for more details."
 )
 warnings.warn(deprecation_warning, DeprecationWarning, stacklevel=2)
 from ignite.metrics.regression.median_absolute_percentage_error import MedianAbsolutePercentageError

ignite/contrib/metrics/regression/median_relative_absolute_error.py

@@ -9,9 +9,8 @@ import warnings
 removed_in = "0.6.0"
 deprecation_warning = (
     f"{__file__} has been moved to ignite/metrics/regression/median_relative_absolute_error.py"
-    f" and will be removed in version {removed_in}"
-    if removed_in
-    else "" ".\n Please refer to the documentation for more details."
+    + f" and will be removed in version {removed_in}"
+    + ".\n Please refer to the documentation for more details."
 )
 warnings.warn(deprecation_warning, DeprecationWarning, stacklevel=2)
 from ignite.metrics.regression.median_relative_absolute_error import MedianRelativeAbsoluteError

ignite/contrib/metrics/regression/r2_score.py

@@ -9,9 +9,8 @@ import warnings
 removed_in = "0.6.0"
 deprecation_warning = (
     f"{__file__} has been moved to ignite/metrics/regression/r2_score.py"
-    f" and will be removed in version {removed_in}"
-    if removed_in
-    else "" ".\n Please refer to the documentation for more details."
+    + f" and will be removed in version {removed_in}"
+    + ".\n Please refer to the documentation for more details."
 )
 warnings.warn(deprecation_warning, DeprecationWarning, stacklevel=2)
 from ignite.metrics.regression.r2_score import R2Score

ignite/contrib/metrics/regression/wave_hedges_distance.py

@@ -9,9 +9,8 @@ import warnings
 removed_in = "0.6.0"
 deprecation_warning = (
     f"{__file__} has been moved to ignite/metrics/regression/wave_hedges_distance.py"
-    f" and will be removed in version {removed_in}"
-    if removed_in
-    else "" ".\n Please refer to the documentation for more details."
+    + f" and will be removed in version {removed_in}"
+    + ".\n Please refer to the documentation for more details."
 )
 warnings.warn(deprecation_warning, DeprecationWarning, stacklevel=2)
 from ignite.metrics.regression.wave_hedges_distance import WaveHedgesDistance

ignite/contrib/metrics/roc_auc.py

@@ -9,7 +9,7 @@ import warnings
 removed_in = "0.6.0"
 deprecation_warning = (
     f"{__file__} has been moved to ignite/metrics/roc_auc.py"
-    + (f" and will be removed in version {removed_in}" if removed_in else "")
+    + f" and will be removed in version {removed_in}"
     + ".\n Please refer to the documentation for more details."
 )
 warnings.warn(deprecation_warning, DeprecationWarning, stacklevel=2)

ignite/distributed/auto.py

@@ -336,6 +336,7 @@ if idist.has_xla_support:
         # From pytorch/xla if `torch_xla.distributed.parallel_loader.MpDeviceLoader` is not available
         def __init__(self, loader: Any, device: torch.device, **kwargs: Any) -> None:
             self._loader = loader
+            # pyrefly: ignore [read-only]
             self._device = device
             self._parallel_loader_kwargs = kwargs
 

ignite/distributed/comp_models/horovod.py

@@ -1,6 +1,6 @@
 import os
 import warnings
-from typing import Any, Callable, cast, List, Mapping, Optional, Tuple
+from typing import Any, Callable, cast, List, Mapping, Optional, Tuple, TYPE_CHECKING
 
 import torch
 
@@ -20,6 +20,11 @@ try:
 except ImportError:
     has_hvd_support = False
 
+    if TYPE_CHECKING:
+        # Tell the type checker that hvd imports are always defined.
+        import horovod.torch as hvd
+        from horovod import run as hvd_mp_spawn
+
 
 if has_hvd_support:
     HOROVOD = "horovod"
@@ -70,6 +75,7 @@ if has_hvd_support:
                 self._init_from_context()
 
         def _create_from_backend(self, backend: str, **kwargs: Any) -> None:
+            # pyrefly: ignore [bad-override]
             self._backend: str = backend
             comm = kwargs.get("comm", None)
             hvd.init(comm=comm)
@@ -129,6 +135,7 @@ if has_hvd_support:
             finalize()
 
         @staticmethod
+        # pyrefly: ignore [bad-override]
         def spawn(
             fn: Callable,
             args: Tuple,

ignite/distributed/comp_models/native.py

@@ -345,6 +345,7 @@ if has_native_dist_support:
             os.environ.update(copy_env_vars)
 
         @staticmethod
+        # pyrefly: ignore [bad-override]
         def spawn(
             fn: Callable,
             args: Tuple,

ignite/distributed/comp_models/xla.py

@@ -54,6 +54,7 @@ if has_xla_support:
         def _create_from_backend(self, backend: str, **kwargs: Any) -> None:
             xm.rendezvous("init")
 
+            # pyrefly: ignore [bad-override]
             self._backend: str = backend
             self._setup_attrs()
 
@@ -106,6 +107,7 @@ if has_xla_support:
             finalize()
 
         @staticmethod
+        # pyrefly: ignore [bad-override]
         def spawn(
             fn: Callable,
             args: Tuple,

ignite/distributed/launcher.py

@@ -322,19 +322,15 @@ class Parallel:
             idist.initialize(self.backend, init_method=self.init_method)
 
         # The logger can be setup from now since idist.initialize() has been called (if needed)
-        self._logger = setup_logger(__name__ + "." + self.__class__.__name__)  # type: ignore[assignment]
+        self._logger = setup_logger(__name__ + "." + self.__class__.__name__)
 
         if self.backend is not None:
             if self._spawn_params is None:
-                self._logger.info(  # type: ignore[attr-defined]
-                    f"Initialized processing group with backend: '{self.backend}'"
-                )
+                self._logger.info(f"Initialized processing group with backend: '{self.backend}'")
             else:
-                self._logger.info(  # type: ignore[attr-defined]
-                    f"Initialized distributed launcher with backend: '{self.backend}'"
-                )
+                self._logger.info(f"Initialized distributed launcher with backend: '{self.backend}'")
                 msg = "\n\t".join([f"{k}: {v}" for k, v in self._spawn_params.items() if v is not None])
-                self._logger.info(f"- Parameters to spawn processes: \n\t{msg}")  # type: ignore[attr-defined]
+                self._logger.info(f"- Parameters to spawn processes: \n\t{msg}")
 
         return self
 

ignite/engine/deterministic.py

@@ -85,7 +85,7 @@ class ReproducibleBatchSampler(BatchSampler):
 
 
 def _get_rng_states() -> List[Any]:
-    output = [random.getstate(), torch.get_rng_state()]
+    output: List[Any] = [random.getstate(), torch.get_rng_state()]
     try:
         import numpy as np
 

ignite/engine/engine.py

@@ -148,7 +148,7 @@ class Engine(Serializable):
         self.should_interrupt = False
         self.state = State()
         self._state_dict_user_keys: List[str] = []
-        self._allowed_events: List[EventEnum] = []
+        self._allowed_events: List[Union[str, EventEnum]] = []
 
         self._dataloader_iter: Optional[Iterator[Any]] = None
         self._init_iter: Optional[int] = None
@@ -163,9 +163,7 @@ class Engine(Serializable):
         # generator provided by self._internal_run_as_gen
         self._internal_run_generator: Optional[Generator[Any, None, State]] = None
 
-    def register_events(
-        self, *event_names: Union[List[str], List[EventEnum]], event_to_attr: Optional[dict] = None
-    ) -> None:
+    def register_events(self, *event_names: Union[str, EventEnum], event_to_attr: Optional[dict] = None) -> None:
         """Add events that can be fired.
 
         Registering an event will let the user trigger these events at any point.
@@ -450,7 +448,7 @@ class Engine(Serializable):
                 first, others = ((resolved_engine,), args[1:])
             else:
                 # metrics do not provide engine when registered
-                first, others = (tuple(), args)  # type: ignore[assignment]
+                first, others = (tuple(), args)
 
             func(*first, *(event_args + others), **kwargs)
 
@@ -989,9 +987,9 @@ class Engine(Serializable):
     def _internal_run_as_gen(self) -> Generator[Any, None, State]:
         self.should_terminate = self.should_terminate_single_epoch = self.should_interrupt = False
         self._init_timers(self.state)
+        start_time = time.time()
         try:
             try:
-                start_time = time.time()
                 self._fire_event(Events.STARTED)
                 yield from self._maybe_terminate_or_interrupt()
 
@@ -1010,7 +1008,7 @@ class Engine(Serializable):
                     # time is available for handlers but must be updated after fire
                     self.state.times[Events.EPOCH_COMPLETED.name] = epoch_time_taken
 
-                    if self.should_terminate_single_epoch != "skip_epoch_completed":  # type: ignore[comparison-overlap]
+                    if self.should_terminate_single_epoch != "skip_epoch_completed":
                         handlers_start_time = time.time()
                         self._fire_event(Events.EPOCH_COMPLETED)
                         epoch_time_taken += time.time() - handlers_start_time
@@ -1043,7 +1041,7 @@ class Engine(Serializable):
             self.state.times[Events.COMPLETED.name] = time_taken
 
             # do not fire Events.COMPLETED if we terminated the run with flag `skip_completed=True`
-            if self.should_terminate != "skip_completed":  # type: ignore[comparison-overlap]
+            if self.should_terminate != "skip_completed":
                 handlers_start_time = time.time()
                 self._fire_event(Events.COMPLETED)
                 time_taken += time.time() - handlers_start_time
@@ -1189,9 +1187,9 @@ class Engine(Serializable):
         # internal_run without generator for BC
         self.should_terminate = self.should_terminate_single_epoch = self.should_interrupt = False
         self._init_timers(self.state)
+        start_time = time.time()
         try:
             try:
-                start_time = time.time()
                 self._fire_event(Events.STARTED)
                 self._maybe_terminate_legacy()
 
@@ -1210,7 +1208,7 @@ class Engine(Serializable):
                     # time is available for handlers but must be updated after fire
                     self.state.times[Events.EPOCH_COMPLETED.name] = epoch_time_taken
 
-                    if self.should_terminate_single_epoch != "skip_epoch_completed":  # type: ignore[comparison-overlap]
+                    if self.should_terminate_single_epoch != "skip_epoch_completed":
                         handlers_start_time = time.time()
                         self._fire_event(Events.EPOCH_COMPLETED)
                         epoch_time_taken += time.time() - handlers_start_time
@@ -1243,7 +1241,7 @@ class Engine(Serializable):
             self.state.times[Events.COMPLETED.name] = time_taken
 
             # do not fire Events.COMPLETED if we terminated the run with flag `skip_completed=True`
-            if self.should_terminate != "skip_completed":  # type: ignore[comparison-overlap]
+            if self.should_terminate != "skip_completed":
                 handlers_start_time = time.time()
                 self._fire_event(Events.COMPLETED)
                 time_taken += time.time() - handlers_start_time

ignite/engine/events.py

@@ -91,7 +91,7 @@ class CallableEventWithFilter:
             raise ValueError("Argument every should be integer and greater than zero")
 
         if once is not None:
-            c1 = isinstance(once, numbers.Integral) and once > 0
+            c1 = isinstance(once, int) and once > 0
             c2 = isinstance(once, Sequence) and len(once) > 0 and all(isinstance(e, int) and e > 0 for e in once)
             if not (c1 or c2):
                 raise ValueError(
@@ -240,6 +240,7 @@ class EventEnum(CallableEventWithFilter, Enum):
     def __new__(cls, value: str) -> "EventEnum":
         obj = CallableEventWithFilter.__new__(cls)
         obj._value_ = value
+        # pyrefly: ignore [bad-return]
         return obj
 
 

ignite/handlers/checkpoint.py

@@ -315,7 +315,7 @@ class Checkpoint(Serializable):
     """
 
     SAVED_CHECKPOINT = CheckpointEvents.SAVED_CHECKPOINT
-    Item = NamedTuple("Item", [("priority", int), ("filename", str)])
+    Item = NamedTuple("Item", [("priority", Union[int, float]), ("filename", str)])
     _state_dict_all_req_keys = ("_saved",)
 
     def __init__(
@@ -323,7 +323,7 @@ class Checkpoint(Serializable):
         to_save: Mapping,
         save_handler: Union[str, Path, Callable, BaseSaveHandler],
         filename_prefix: str = "",
-        score_function: Optional[Callable] = None,
+        score_function: Optional[Callable[[Engine], Union[int, float]]] = None,
         score_name: Optional[str] = None,
         n_saved: Union[int, None] = 1,
         global_step_transform: Optional[Callable] = None,