Refactor - cleanup of old stuff and embarasing typos

CHANGELOG.md

@@ -7,6 +7,29 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 ## [Unreleased]
 
+## [0.6.1] - 2026-05-05
+
+Internal cleanup pass and a few small breaking removals. For older Python or
+older import paths, pin a previous `livelossplot` version.
+
+### Removed
+
+- The `livelossplot.keras` / `livelossplot.tf_keras` / `livelossplot.poutyne` /
+  `livelossplot.pytorch_ignite` import shortcuts (deprecated since 0.5.0).
+  Use `livelossplot.inputs.*` or `from livelossplot import PlotLossesKeras` etc.
+- `PlotLosses.draw()` (deprecated since 0.5.0). Use `.send()`.
+
+### Changed
+
+- `ExtremaPrinter` parameter `massage_template` renamed to `message_template`.
+- `Plot2d` parameter `valiation_data` renamed to `validation_data`.
+
+### Fixed
+
+- Mutable default arguments in `PlotLosses(outputs=...)` and `MatplotlibPlot(extra_plots=...)`.
+- `MainLogger.groups = None` now correctly resets to `{}`.
+- `ipython` declared as a runtime dependency.
+
 ## [0.6.0] - 2026-05-04
 
 ### Changed

examples/2d_prediction_maps.ipynb

@@ -233,7 +233,7 @@
         "model = MLP(6)\n",
         "\n",
         "plot2d = matplotlib_subplots.Plot2d(model, X_train, y_train,\n",
-        "                                    valiation_data=(X_test, y_test),\n",
+        "                                    validation_data=(X_test, y_test),\n",
         "                                    margin=0.2, h=0.02, device=device)\n",
         "plot2d.predict = plot2d._predict_pytorch\n",
         "liveloss = PlotLosses(outputs=[plot2d])\n",
@@ -272,7 +272,7 @@
         "model = MLP(3, activation=nn.Sigmoid())\n",
         "\n",
         "plot2d = matplotlib_subplots.Plot2d(model, X_train, y_train,\n",
-        "                                    valiation_data=(X_test, y_test),\n",
+        "                                    validation_data=(X_test, y_test),\n",
         "                                    margin=0.2, h=0.02, device=device)\n",
         "plot2d.predict = plot2d._predict_pytorch\n",
         "liveloss = PlotLosses(outputs=[plot2d])\n",

livelossplot/__init__.py

@@ -2,50 +2,20 @@
 .. include:: ../README.md
 """
 
-import sys
-import warnings
-from importlib.util import find_spec
-
+from . import inputs, outputs
+from .inputs import *
 from .main_logger import MainLogger
 from .plot_losses import PlotLosses
-from . import inputs
-from .inputs import *
-from . import outputs
 from .version import __version__
 
-_input_plugin_dict = {
-    'keras': 'Keras',
-    'tf_keras': 'KerasTF',
-    'pytorch_ignite': 'Ignite',
-    'poutyne': 'Poutyne',
-}
-
-
-class OldDependenciesFinder:
-    """
-    Data package module loader finder. This class sits on `sys.meta_path` and returns the
-    loader it knows for a given path, if it knows a compatible loader.
-    """
-    @classmethod
-    def find_spec(self, fullname: str, *_, **__):
-        """This functions is what gets executed by the loader.
-        Args:
-            fullname: name of the called module
-        """
-        parts = fullname.split('.')
-        if len(parts) == 2 and parts[0] == 'livelossplot' and parts[1] in _input_plugin_dict:
-            name = parts[1]
-            msg = 'livelossplot.{name} will be deprecated, please use livelossplot.inputs.{name}\n'
-            msg += 'or use callback directly: from livelossplot import PlotLosses{new_name}'
-            warnings.warn(msg.format(name=name, new_name=_input_plugin_dict[name]), DeprecationWarning)
-            fullname = 'livelossplot.inputs.{name}'.format(name=name)
-            return find_spec(fullname)
-        return None
-
-
-sys.meta_path.append(OldDependenciesFinder())
-
 __all__ = [
-    'MainLogger', 'inputs', 'outputs', 'PlotLosses', 'PlotLossesKeras', 'PlotLossesKerasTF', 'PlotLossesIgnite',
-    'PlotLossesPoutyne'
+    'MainLogger',
+    'PlotLosses',
+    'PlotLossesIgnite',
+    'PlotLossesKeras',
+    'PlotLossesKerasTF',
+    'PlotLossesPoutyne',
+    '__version__',
+    'inputs',
+    'outputs',
 ]

livelossplot/inputs/generic_keras.py

@@ -1,4 +1,3 @@
-from typing import Dict
 
 from livelossplot.plot_losses import PlotLosses
 
@@ -12,7 +11,7 @@ def __init__(self, **kwargs):
         """
         self.liveplot = PlotLosses(**kwargs)
 
-    def on_epoch_end(self, epoch: int, logs: Dict[str, float]):
+    def on_epoch_end(self, epoch: int, logs: dict[str, float]):
         """Send metrics to livelossplot
         Args:
             epoch: epoch number

livelossplot/inputs/poutyne.py

@@ -1,6 +1,6 @@
-from typing import Dict
 
 from poutyne.framework import Callback
+
 from ..plot_losses import PlotLosses
 
 
@@ -11,7 +11,7 @@ def __init__(self, **kwargs):
         Args:
             **kwargs: keyword arguments that will be passed to PlotLosses constructor
         """
-        super(PlotLossesCallback, self).__init__()
+        super().__init__()
         self.liveplot = PlotLosses(**kwargs)
         self.metrics = None
 
@@ -21,7 +21,7 @@ def on_train_begin(self, logs):
         self.metrics = list(metrics)
         self.metrics += ['val_' + metric for metric in metrics]
 
-    def on_epoch_end(self, epoch: int, logs: Dict[str, float]):
+    def on_epoch_end(self, epoch: int, logs: dict[str, float]):
         """Send metrics to livelossplot
         Args:
             epoch: epoch number

livelossplot/inputs/pytorch_ignite.py

@@ -1,12 +1,12 @@
-from typing import Optional
 
 import ignite.engine
 from ignite.handlers import global_step_from_engine
+
 from livelossplot.plot_losses import PlotLosses
 
 
 class PlotLossesCallback:
-    def __init__(self, train_engine: Optional[ignite.engine.Engine] = None, **kwargs):
+    def __init__(self, train_engine: ignite.engine.Engine | None = None, **kwargs):
         """
         Args:
             train_engine: engine with global step information, send metohod callback will be attached to it
@@ -51,5 +51,5 @@ def store(self, engine: ignite.engine.Engine):
         if not self.train_engine:
             self.send()
 
-    def send(self, _: Optional[ignite.engine.Engine] = None):
+    def send(self, _: ignite.engine.Engine | None = None):
         self.liveplot.send()

livelossplot/inputs/tf_keras.py

@@ -1,4 +1,5 @@
 from tensorflow import keras
+
 from .generic_keras import _PlotLossesCallback