pytorch-ignite 0.6.0.dev20260118__py3-none-any.whl → 0.6.0.dev20260120__py3-none-any.whl

ignite/__init__.py

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

ignite/handlers/tensorboard_logger.py

@@ -1,6 +1,6 @@
 """TensorBoard logger and its helper handlers."""
 
-from typing import Any, Callable, List, Optional, Union
+from typing import Any, Callable
 
 from torch.optim import Optimizer
 
@@ -321,16 +321,16 @@ class OutputHandler(BaseOutputHandler):
     def __init__(
         self,
         tag: str,
-        metric_names: Optional[Union[List[str], str]] = None,
-        output_transform: Optional[Callable] = None,
-        global_step_transform: Optional[Callable[[Engine, Union[str, Events]], int]] = None,
-        state_attributes: Optional[List[str]] = None,
+        metric_names: list[str] | str | None = None,
+        output_transform: Callable | None = None,
+        global_step_transform: Callable[[Engine, str | Events], int] | None = None,
+        state_attributes: list[str] | None = None,
     ):
         super(OutputHandler, self).__init__(
             tag, metric_names, output_transform, global_step_transform, state_attributes
         )
 
-    def __call__(self, engine: Engine, logger: TensorboardLogger, event_name: Union[str, Events]) -> None:
+    def __call__(self, engine: Engine, logger: TensorboardLogger, event_name: str | Events) -> None:
         if not isinstance(logger, TensorboardLogger):
             raise RuntimeError("Handler 'OutputHandler' works only with TensorboardLogger")
 
@@ -377,10 +377,10 @@ class OptimizerParamsHandler(BaseOptimizerParamsHandler):
             )
     """
 
-    def __init__(self, optimizer: Optimizer, param_name: str = "lr", tag: Optional[str] = None):
+    def __init__(self, optimizer: Optimizer, param_name: str = "lr", tag: str | None = None):
         super(OptimizerParamsHandler, self).__init__(optimizer, param_name, tag)
 
-    def __call__(self, engine: Engine, logger: TensorboardLogger, event_name: Union[str, Events]) -> None:
+    def __call__(self, engine: Engine, logger: TensorboardLogger, event_name: str | Events) -> None:
         if not isinstance(logger, TensorboardLogger):
             raise RuntimeError("Handler OptimizerParamsHandler works only with TensorboardLogger")
 
@@ -463,7 +463,7 @@ class WeightsScalarHandler(BaseWeightsScalarHandler):
         optional argument `whitelist` added.
     """
 
-    def __call__(self, engine: Engine, logger: TensorboardLogger, event_name: Union[str, Events]) -> None:
+    def __call__(self, engine: Engine, logger: TensorboardLogger, event_name: str | Events) -> None:
         if not isinstance(logger, TensorboardLogger):
             raise RuntimeError("Handler 'WeightsScalarHandler' works only with TensorboardLogger")
 
@@ -542,7 +542,7 @@ class WeightsHistHandler(BaseWeightsHandler):
         optional argument `whitelist` added.
     """
 
-    def __call__(self, engine: Engine, logger: TensorboardLogger, event_name: Union[str, Events]) -> None:
+    def __call__(self, engine: Engine, logger: TensorboardLogger, event_name: str | Events) -> None:
         if not isinstance(logger, TensorboardLogger):
             raise RuntimeError("Handler 'WeightsHistHandler' works only with TensorboardLogger")
 
@@ -624,7 +624,7 @@ class GradsScalarHandler(BaseWeightsScalarHandler):
         optional argument `whitelist` added.
     """
 
-    def __call__(self, engine: Engine, logger: TensorboardLogger, event_name: Union[str, Events]) -> None:
+    def __call__(self, engine: Engine, logger: TensorboardLogger, event_name: str | Events) -> None:
         if not isinstance(logger, TensorboardLogger):
             raise RuntimeError("Handler 'GradsScalarHandler' works only with TensorboardLogger")
 
@@ -701,7 +701,7 @@ class GradsHistHandler(BaseWeightsHandler):
         optional argument `whitelist` added.
     """
 
-    def __call__(self, engine: Engine, logger: TensorboardLogger, event_name: Union[str, Events]) -> None:
+    def __call__(self, engine: Engine, logger: TensorboardLogger, event_name: str | Events) -> None:
         if not isinstance(logger, TensorboardLogger):
             raise RuntimeError("Handler 'GradsHistHandler' works only with TensorboardLogger")
 

ignite/metrics/accuracy.py

@@ -1,4 +1,4 @@
-from typing import Callable, Optional, Sequence, Tuple, Union
+from typing import Callable, Sequence
 
 import torch
 
@@ -13,12 +13,12 @@ class _BaseClassification(Metric):
         self,
         output_transform: Callable = lambda x: x,
         is_multilabel: bool = False,
-        device: Union[str, torch.device] = torch.device("cpu"),
+        device: str | torch.device = torch.device("cpu"),
         skip_unrolling: bool = False,
     ):
         self._is_multilabel = is_multilabel
-        self._type: Optional[str] = None
-        self._num_classes: Optional[int] = None
+        self._type: str | None = None
+        self._num_classes: int | None = None
         super(_BaseClassification, self).__init__(
             output_transform=output_transform, device=device, skip_unrolling=skip_unrolling
         )
@@ -38,7 +38,7 @@ class _BaseClassification(Metric):
             )
 
         y_shape = y.shape
-        y_pred_shape: Tuple[int, ...] = y_pred.shape
+        y_pred_shape: tuple[int, ...] = y_pred.shape
 
         if y.ndimension() + 1 == y_pred.ndimension():
             y_pred_shape = (y_pred_shape[0],) + y_pred_shape[2:]
@@ -223,7 +223,7 @@ class Accuracy(_BaseClassification):
         self,
         output_transform: Callable = lambda x: x,
         is_multilabel: bool = False,
-        device: Union[str, torch.device] = torch.device("cpu"),
+        device: str | torch.device = torch.device("cpu"),
         skip_unrolling: bool = False,
     ):
         super(Accuracy, self).__init__(

ignite/metrics/cohen_kappa.py

@@ -1,4 +1,4 @@
-from typing import Callable, Optional, Union
+from typing import Callable, Literal
 
 import torch
 
@@ -56,9 +56,9 @@ class CohenKappa(EpochMetric):
     def __init__(
         self,
         output_transform: Callable = lambda x: x,
-        weights: Optional[str] = None,
+        weights: Literal["linear", "quadratic"] | None = None,
         check_compute_fn: bool = False,
-        device: Union[str, torch.device] = torch.device("cpu"),
+        device: str | torch.device = torch.device("cpu"),
         skip_unrolling: bool = False,
     ):
         try:
@@ -68,8 +68,8 @@ class CohenKappa(EpochMetric):
         if weights not in (None, "linear", "quadratic"):
             raise ValueError("Kappa Weighting type must be None or linear or quadratic.")
 
-        # initalize weights
-        self.weights = weights
+        # initialize weights
+        self.weights: Literal["linear", "quadratic"] | None = weights
 
         super(CohenKappa, self).__init__(
             self._cohen_kappa_score,

ignite/metrics/metric_group.py

@@ -1,4 +1,4 @@
-from typing import Any, Callable, Dict, Sequence, Tuple
+from typing import Any, Callable, Sequence
 
 import torch
 
@@ -44,10 +44,10 @@ class MetricGroup(Metric):
         ``skip_unrolling`` argument is added.
     """
 
-    _state_dict_all_req_keys: Tuple[str, ...] = ("metrics",)
+    _state_dict_all_req_keys: tuple[str, ...] = ("metrics",)
 
     def __init__(
-        self, metrics: Dict[str, Metric], output_transform: Callable = lambda x: x, skip_unrolling: bool = False
+        self, metrics: dict[str, Metric], output_transform: Callable = lambda x: x, skip_unrolling: bool = False
     ):
         self.metrics = metrics
         super(MetricGroup, self).__init__(output_transform=output_transform, skip_unrolling=skip_unrolling)
@@ -60,5 +60,5 @@ class MetricGroup(Metric):
         for m in self.metrics.values():
             m.update(m._output_transform(output))
 
-    def compute(self) -> Dict[str, Any]:
+    def compute(self) -> dict[str, Any]:
         return {k: m.compute() for k, m in self.metrics.items()}

{pytorch_ignite-0.6.0.dev20260118.dist-info → pytorch_ignite-0.6.0.dev20260120.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pytorch-ignite
-Version: 0.6.0.dev20260118
+Version: 0.6.0.dev20260120
 Summary: A lightweight library to help with training neural networks in PyTorch.
 Project-URL: Homepage, https://pytorch-ignite.ai
 Project-URL: Repository, https://github.com/pytorch/ignite

{pytorch_ignite-0.6.0.dev20260118.dist-info → pytorch_ignite-0.6.0.dev20260120.dist-info}/RECORD

@@ -1,4 +1,4 @@
-ignite/__init__.py,sha256=xQh94zBR0e8L-_Niygli9Iy2moUBK8XfRHQ_J9miR1Q,194
+ignite/__init__.py,sha256=KCPXiPhh4fXqJPVd1LqDaY1tRXNOBFLJoL2qhYTHDOs,194
 ignite/_utils.py,sha256=XDPpUDJ8ykLXWMV2AYTqGSj8XCfApsyzsQ3Vij_OB4M,182
 ignite/exceptions.py,sha256=5ZWCVLPC9rgoW8t84D-VeEleqz5O7XpAGPpCdU8rKd0,150
 ignite/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -73,7 +73,7 @@ ignite/handlers/param_scheduler.py,sha256=Tn4o27YBrp5JsuadHobIrsHfmvB_cR1IrV_oV1
 ignite/handlers/polyaxon_logger.py,sha256=7nyOhu4rgg4Sj6uTNUobPG9IJqYxjb_fopKiF4fk9vc,12346
 ignite/handlers/state_param_scheduler.py,sha256=B89YKZyj9DXLXQyr3amDNMslUOWNHZDis2DXIwW0q10,20841
 ignite/handlers/stores.py,sha256=8XM_Qqsitfu0WtOOE-K2FMtv51vD90r3GgQlCzRABYc,2616
-ignite/handlers/tensorboard_logger.py,sha256=e6gS9b7dkGjm-iR4iTQWPZcYKrDUml7cmrriNXoMBoU,27959
+ignite/handlers/tensorboard_logger.py,sha256=_kESDcCqFqhdA-IIgwVnMZ3JxS1O7grROG5Bc6wRvoc,27873
 ignite/handlers/terminate_on_nan.py,sha256=RFSKd3Oqn9Me2xLCos4lSE-hnY7fYWWjE9blioeMlIs,2103
 ignite/handlers/time_limit.py,sha256=heTuS-ReBbOUCm1NcNJGhzxI080Hanc4hOLB2Y4GeZk,1567
 ignite/handlers/time_profilers.py,sha256=8iCcBYPxv0vKFSO_ujFV0ST54a9PD9ezFLvYTIu9lFI,30482
@@ -84,10 +84,10 @@ ignite/handlers/visdom_logger.py,sha256=OlnqVDXKYTea6VG7rwcQdzrfYmai0SSxNLh2Kqsj
 ignite/handlers/wandb_logger.py,sha256=9HbwRMHzWckrZ-m0rkMF5Ug6r9C9J4sdq73yqaAHWYE,14829
 ignite/metrics/__init__.py,sha256=m-8F8J17r-aEwsO6Ww-8AqDRN59WFfYBwCDKwqGDSmI,3627
 ignite/metrics/accumulation.py,sha256=xWdsm9u6JfsfODX_GUKzQc_omrdFDJ4yELBR-xXgc4s,12448
-ignite/metrics/accuracy.py,sha256=W8mO4W11VzryMXKy8G7W_g4A9PH9RYpejW_tQ-T_Txw,10245
+ignite/metrics/accuracy.py,sha256=hVDvMG2kc-EN9H54tdMYWy6D3Zj74DzpHdG3Eq5B6Nc,10203
 ignite/metrics/average_precision.py,sha256=laDD8BnAC5OuAJrCRtwCZ7EjjoQKRb7D3o-86IRsdN4,3681
 ignite/metrics/classification_report.py,sha256=zjGlaMnRz2__op6hrZq74OusO0W_5B1AIe8KzYGFilM,5988
-ignite/metrics/cohen_kappa.py,sha256=Qwcd4P2kN12CVCFC-kVdzn_2XV7kGzP6LlWkK209JJ8,3815
+ignite/metrics/cohen_kappa.py,sha256=D3vOkIK86qoGvg9nis5PlbRJ8uYZmNdpLJaBRVi0Moc,3865
 ignite/metrics/confusion_matrix.py,sha256=dZDuK3vxrrbiQh6VfyV5aWFpuTJWsfnZ30Mxt6u6eOA,18215
 ignite/metrics/cosine_similarity.py,sha256=9f5dM0QaiXBznidlUuKb8Q3E45W_Z_hGiy9WP5Fpcqw,4416
 ignite/metrics/entropy.py,sha256=gJZkR5Sl1ZdIzJ9pFkydf1186bZU8OnkOLvOtKz6Wrs,4511
@@ -105,7 +105,7 @@ ignite/metrics/mean_average_precision.py,sha256=cXP9pYidQnAazGXBrhC80WoI4eK4lb3a
 ignite/metrics/mean_pairwise_distance.py,sha256=Ys6Rns6s-USS_tyP6Pa3bWZSI7f_hP5-lZM64UGJGjo,4104
 ignite/metrics/mean_squared_error.py,sha256=UnLLb7XKwvHhOxQWTVhYDCluKETVazG0yDSdX4s9pQY,3666
 ignite/metrics/metric.py,sha256=T3IiFIGTv_UOScd8ei4H9SraHfTJ09OM8I6hRfzr_sA,35141
-ignite/metrics/metric_group.py,sha256=UE7WrMbpKlO9_DPqxQdlmFAWveWoT1knKwRlHDl9YIU,2544
+ignite/metrics/metric_group.py,sha256=yiS7MXQB1wROUXnK57omu6oGMkQ0WudYDx-Ee-togmo,2531
 ignite/metrics/metrics_lambda.py,sha256=NwKZ1J-KzFFbSw7YUaNJozdfKZLVqrkjQvFKT6ixnkg,7309
 ignite/metrics/multilabel_confusion_matrix.py,sha256=1pjLNPGTDJWAkN_BHdBPekcish6Ra0uRUeEbdj3Dm6Y,7377
 ignite/metrics/mutual_information.py,sha256=lu1ucVfkx01tGQfELyXzS9woCPOMVImFHfrbIXCvPe8,4692
@@ -153,7 +153,7 @@ ignite/metrics/regression/spearman_correlation.py,sha256=IzmN4WIe7C4cTUU3BOkBmaw
 ignite/metrics/regression/wave_hedges_distance.py,sha256=Ji_NRUgnZ3lJgi5fyNFLRjbHO648z4dBmqVDQU9ImKA,2792
 ignite/metrics/vision/__init__.py,sha256=lPBAEq1idc6Q17poFm1SjttE27irHF1-uNeiwrxnLrU,159
 ignite/metrics/vision/object_detection_average_precision_recall.py,sha256=4wwiNVd658ynIpIbQlffTA-ehvyJ2EzmJ5pBSBuA8XQ,25091
-pytorch_ignite-0.6.0.dev20260118.dist-info/METADATA,sha256=EhmYgR6EJCry4dtHlWqk0E-DvLRY3sphi_JZ_eJfhnE,27979
-pytorch_ignite-0.6.0.dev20260118.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-pytorch_ignite-0.6.0.dev20260118.dist-info/licenses/LICENSE,sha256=SwJvaRmy1ql-k9_nL4WnER4_ODTMF9fWoP9HXkoicgw,1527
-pytorch_ignite-0.6.0.dev20260118.dist-info/RECORD,,
+pytorch_ignite-0.6.0.dev20260120.dist-info/METADATA,sha256=s3fEGt1qbwRV18P-TNhk7LzNiE9cQPacXI8Mey3Cztk,27979
+pytorch_ignite-0.6.0.dev20260120.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+pytorch_ignite-0.6.0.dev20260120.dist-info/licenses/LICENSE,sha256=SwJvaRmy1ql-k9_nL4WnER4_ODTMF9fWoP9HXkoicgw,1527
+pytorch_ignite-0.6.0.dev20260120.dist-info/RECORD,,