pytorch-ignite 0.6.0.dev20260109__py3-none-any.whl → 0.6.0.dev20260111__py3-none-any.whl

ignite/__init__.py

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

ignite/handlers/clearml_logger.py

@@ -325,7 +325,7 @@ class OutputHandler(BaseOutputHandler):
     def __init__(
         self,
         tag: str,
-        metric_names: Optional[List[str]] = None,
+        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,
@@ -344,8 +344,7 @@ class OutputHandler(BaseOutputHandler):
 
         if not isinstance(global_step, int):
             raise TypeError(
-                f"global_step must be int, got {type(global_step)}."
-                " Please check the output of global_step_transform."
+                f"global_step must be int, got {type(global_step)}. Please check the output of global_step_transform."
             )
 
         for key, value in metrics.items():

ignite/handlers/lr_finder.py

@@ -317,7 +317,7 @@ class FastaiLRFinder:
             lrs = lrs[skip_start:-skip_end]
             losses = losses[skip_start:-skip_end]
 
-        plt.legend(legends)
+        plt.legend(legends)  # pyrefly: ignore
         # Plot loss as a function of the learning rate
         ax.plot(lrs, losses)
         if log_lr:

ignite/handlers/polyaxon_logger.py

@@ -5,7 +5,6 @@ from typing import Any, Callable, List, Optional, Union
 from torch.optim import Optimizer
 
 from ignite.engine import Engine, Events
-
 from ignite.handlers.base_logger import BaseLogger, BaseOptimizerParamsHandler, BaseOutputHandler
 from ignite.handlers.utils import global_step_from_engine  # noqa
 
@@ -237,7 +236,7 @@ class OutputHandler(BaseOutputHandler):
     def __init__(
         self,
         tag: str,
-        metric_names: Optional[List[str]] = None,
+        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,
@@ -256,8 +255,7 @@ class OutputHandler(BaseOutputHandler):
 
         if not isinstance(global_step, int):
             raise TypeError(
-                f"global_step must be int, got {type(global_step)}."
-                " Please check the output of global_step_transform."
+                f"global_step must be int, got {type(global_step)}. Please check the output of global_step_transform."
             )
 
         metrics.update({"step": global_step})

ignite/handlers/tensorboard_logger.py

@@ -5,7 +5,6 @@ from typing import Any, Callable, List, Optional, Union
 from torch.optim import Optimizer
 
 from ignite.engine import Engine, Events
-
 from ignite.handlers.base_logger import (
     BaseLogger,
     BaseOptimizerParamsHandler,
@@ -322,7 +321,7 @@ class OutputHandler(BaseOutputHandler):
     def __init__(
         self,
         tag: str,
-        metric_names: Optional[List[str]] = None,
+        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,
@@ -340,8 +339,7 @@ class OutputHandler(BaseOutputHandler):
         global_step = self.global_step_transform(engine, event_name)
         if not isinstance(global_step, int):
             raise TypeError(
-                f"global_step must be int, got {type(global_step)}."
-                " Please check the output of global_step_transform."
+                f"global_step must be int, got {type(global_step)}. Please check the output of global_step_transform."
             )
 
         for key, value in metrics.items():

ignite/handlers/visdom_logger.py

@@ -1,7 +1,7 @@
 """Visdom logger and its helper handlers."""
 
 import os
-from typing import Any, Callable, Dict, List, Optional, Union, TYPE_CHECKING
+from typing import TYPE_CHECKING, Any, Callable, Dict, List, Optional, Union
 
 import torch
 import torch.nn as nn
@@ -14,7 +14,6 @@ from ignite.handlers.base_logger import (
     BaseOutputHandler,
     BaseWeightsScalarHandler,
 )
-
 from ignite.handlers.utils import global_step_from_engine  # noqa
 
 __all__ = [
@@ -361,7 +360,7 @@ class OutputHandler(BaseOutputHandler, _BaseVisDrawer):
     def __init__(
         self,
         tag: str,
-        metric_names: Optional[str] = None,
+        metric_names: Optional[Union[List[str], str]] = None,
         output_transform: Optional[Callable] = None,
         global_step_transform: Optional[Callable[[Engine, Union[str, Events]], int]] = None,
         show_legend: bool = False,
@@ -382,8 +381,7 @@ class OutputHandler(BaseOutputHandler, _BaseVisDrawer):
 
         if not isinstance(global_step, int):
             raise TypeError(
-                f"global_step must be int, got {type(global_step)}."
-                " Please check the output of global_step_transform."
+                f"global_step must be int, got {type(global_step)}. Please check the output of global_step_transform."
             )
 
         for key, value in metrics.items():

ignite/handlers/wandb_logger.py

@@ -6,7 +6,6 @@ from warnings import warn
 from torch.optim import Optimizer
 
 from ignite.engine import Engine, Events
-
 from ignite.handlers.base_logger import BaseLogger, BaseOptimizerParamsHandler, BaseOutputHandler
 from ignite.handlers.utils import global_step_from_engine  # noqa
 
@@ -277,7 +276,7 @@ class OutputHandler(BaseOutputHandler):
     def __init__(
         self,
         tag: str,
-        metric_names: Optional[List[str]] = None,
+        metric_names: Optional[Union[List[str], str]] = None,
         output_transform: Optional[Callable] = None,
         global_step_transform: Optional[Callable[[Engine, Union[str, Events]], int]] = None,
         sync: Optional[bool] = None,
@@ -294,8 +293,7 @@ class OutputHandler(BaseOutputHandler):
         global_step = self.global_step_transform(engine, event_name)
         if not isinstance(global_step, int):
             raise TypeError(
-                f"global_step must be int, got {type(global_step)}."
-                " Please check the output of global_step_transform."
+                f"global_step must be int, got {type(global_step)}. Please check the output of global_step_transform."
             )
 
         metrics = self._setup_output_metrics_state_attrs(engine, log_text=True, key_tuple=False)

{pytorch_ignite-0.6.0.dev20260109.dist-info → pytorch_ignite-0.6.0.dev20260111.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pytorch-ignite
-Version: 0.6.0.dev20260109
+Version: 0.6.0.dev20260111
 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.dev20260109.dist-info → pytorch_ignite-0.6.0.dev20260111.dist-info}/RECORD

@@ -1,4 +1,4 @@
-ignite/__init__.py,sha256=1gr3jaj8-9mq5MSVFKOJ9C7hDUoUwFudcoLgPq4ZI_Q,194
+ignite/__init__.py,sha256=MiNYi1mCgbI-dp3ZwOIcFyMc9ZoULQr7MFWApr8c9Ow,194
 ignite/_utils.py,sha256=XDPpUDJ8ykLXWMV2AYTqGSj8XCfApsyzsQ3Vij_OB4M,182
 ignite/exceptions.py,sha256=5ZWCVLPC9rgoW8t84D-VeEleqz5O7XpAGPpCdU8rKd0,150
 ignite/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -62,26 +62,26 @@ ignite/engine/utils.py,sha256=QG5mkdg4OipspqgpNQcJuoHTYdr2Sx5LS16kfjOHDdI,1073
 ignite/handlers/__init__.py,sha256=Qq85YTtHPcii6UAfMOoCPg9RwigH96iqxOJKIlRfDqw,2728
 ignite/handlers/base_logger.py,sha256=wPiGn9iCh5ung1GaRUf_qAlqe63h1NpUUQ0XK709p2k,13011
 ignite/handlers/checkpoint.py,sha256=u6cFUDxAoSSBKCBprmDud2LEZGDEYHvyCoLUmtG3Xd4,46309
-ignite/handlers/clearml_logger.py,sha256=0-57RYznIz-EgTsKtkKFPdGGFQXJIhq146H_qiE8hVc,37897
+ignite/handlers/clearml_logger.py,sha256=MlqHtFjC04yrtvAmFTzPPgwDkiRrXGrVpi1NRAoMqmk,37890
 ignite/handlers/early_stopping.py,sha256=UA6TiKho5CbD085R-16H8w3r0BYPQcWQjhEXg8aITSw,4139
 ignite/handlers/ema_handler.py,sha256=SmUyyWIFPZW3yMvjD_sSk5m_LfnMFl9R-uQdbXNFfY0,11854
 ignite/handlers/fbresearch_logger.py,sha256=MfQeiBIXBYLEwZoDIld2oCceMeTAsz8rc5cd7fLtpJs,11133
-ignite/handlers/lr_finder.py,sha256=EMcQR3NDPOuh2s85a5Zu5Bqt0I4pg1cACJpjSa5cO4A,22100
+ignite/handlers/lr_finder.py,sha256=VEJyUTjEXkJzrSaVzMNIwON5uUIpBGIeGWbNucX_kzk,22119
 ignite/handlers/mlflow_logger.py,sha256=M5Mggrnr2wMsms8wbEaHqNtTk5L1zNs1MlPWD0ZCpDQ,13894
 ignite/handlers/neptune_logger.py,sha256=Rv-O_i0zGZC2Ozzeetxv7rtD7iP3IeWEcbY-U28Mkzg,27348
 ignite/handlers/param_scheduler.py,sha256=Tn4o27YBrp5JsuadHobIrsHfmvB_cR1IrV_oV1Eo7us,68373
-ignite/handlers/polyaxon_logger.py,sha256=5b7Zxhksne8Ufg_SBTG-rlf_9CPSjkBQOJR4-ynoZnQ,12354
+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=q3YxXkbIFayBggI_kcHyl-upttVVjjnqFOLgyjj2cRo,27967
+ignite/handlers/tensorboard_logger.py,sha256=e6gS9b7dkGjm-iR4iTQWPZcYKrDUml7cmrriNXoMBoU,27959
 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
 ignite/handlers/timing.py,sha256=nHeBHvPwYdPRMAx-jk_8MjZit4a7rmsmIWkUrajAG-s,4705
 ignite/handlers/tqdm_logger.py,sha256=3kxH39vM0LCDVwIZl9HQRaWM2Pr6bYC_l9oydFJmdM4,13093
 ignite/handlers/utils.py,sha256=X4LRqo1kqGsbmX0pEuZKYR6K4C8sZudAqxCLriiXtCg,872
-ignite/handlers/visdom_logger.py,sha256=RY5ss3NAPad7d3xFFnqczCtuO6RgmWq9ROz-sFf6imI,21862
-ignite/handlers/wandb_logger.py,sha256=vGok3gADQmTNkc6KkfFBreYoHAO8EneuU65xjBpT5-Q,14837
+ignite/handlers/visdom_logger.py,sha256=OlnqVDXKYTea6VG7rwcQdzrfYmai0SSxNLh2KqsjreU,21860
+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
@@ -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.dev20260109.dist-info/METADATA,sha256=r-rroOFoSErSxxGGo-lndF5NkiaCmmXEf3p5gyX9qfE,27979
-pytorch_ignite-0.6.0.dev20260109.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-pytorch_ignite-0.6.0.dev20260109.dist-info/licenses/LICENSE,sha256=SwJvaRmy1ql-k9_nL4WnER4_ODTMF9fWoP9HXkoicgw,1527
-pytorch_ignite-0.6.0.dev20260109.dist-info/RECORD,,
+pytorch_ignite-0.6.0.dev20260111.dist-info/METADATA,sha256=5xtdAYoZ8V5bblHaFYKlRQ3u415PHC74htcAKkojBQk,27979
+pytorch_ignite-0.6.0.dev20260111.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+pytorch_ignite-0.6.0.dev20260111.dist-info/licenses/LICENSE,sha256=SwJvaRmy1ql-k9_nL4WnER4_ODTMF9fWoP9HXkoicgw,1527
+pytorch_ignite-0.6.0.dev20260111.dist-info/RECORD,,