remove global_step_from_engine from __all__

pytorch · Mar 15, 2024 · 40552c6 · 40552c6
1 parent 961066c
commit 40552c6
Show file tree

Hide file tree

Showing 5 changed files with 2 additions and 5 deletions.
diff --git a/ignite/contrib/handlers/neptune_logger.py b/ignite/contrib/handlers/neptune_logger.py
@@ -30,7 +30,6 @@
     "OutputHandler",
     "WeightsScalarHandler",
     "GradsScalarHandler",
-    "global_step_from_engine",
 ]
 NeptuneLogger = NeptuneLogger
 NeptuneSaver = NeptuneSaver

diff --git a/ignite/contrib/handlers/polyaxon_logger.py b/ignite/contrib/handlers/polyaxon_logger.py
@@ -15,7 +15,7 @@
 from ignite.handlers import global_step_from_engine
 from ignite.handlers.polyaxon_logger import OptimizerParamsHandler, OutputHandler, PolyaxonLogger
 
-__all__ = ["PolyaxonLogger", "OutputHandler", "OptimizerParamsHandler", "global_step_from_engine"]
+__all__ = ["PolyaxonLogger", "OutputHandler", "OptimizerParamsHandler"]
 PolyaxonLogger = PolyaxonLogger
 OutputHandler = OutputHandler
 OptimizerParamsHandler = OptimizerParamsHandler
diff --git a/ignite/contrib/handlers/tensorboard_logger.py b/ignite/contrib/handlers/tensorboard_logger.py
@@ -32,7 +32,6 @@
     "WeightsHistHandler",
     "GradsScalarHandler",
     "GradsHistHandler",
-    "global_step_from_engine",
 ]
 TensorboardLogger = TensorboardLogger
 OptimizerParamsHandler = OptimizerParamsHandler

diff --git a/ignite/contrib/handlers/visdom_logger.py b/ignite/contrib/handlers/visdom_logger.py
@@ -28,7 +28,6 @@
     "OutputHandler",
     "WeightsScalarHandler",
     "GradsScalarHandler",
-    "global_step_from_engine",
 ]
 VisdomLogger = VisdomLogger
 OptimizerParamsHandler = OptimizerParamsHandler

diff --git a/ignite/contrib/handlers/wandb_logger.py b/ignite/contrib/handlers/wandb_logger.py
@@ -15,7 +15,7 @@
 from ignite.handlers import global_step_from_engine
 from ignite.handlers.wandb_logger import OptimizerParamsHandler, OutputHandler, WandBLogger
 
-__all__ = ["WandBLogger", "OutputHandler", "OptimizerParamsHandler", "global_step_from_engine"]
+__all__ = ["WandBLogger", "OutputHandler", "OptimizerParamsHandler"]
 WandBLogger = WandBLogger
 OutputHandler = OutputHandler
 OptimizerParamsHandler = OptimizerParamsHandler