Fix typing of InputData type

Co-authored-by: Roland-djee <[email protected]>

qadence/ml_tools/printing.py

@@ -16,7 +16,7 @@
 logger = getLogger(__name__)
 
 PlottingFunction = Callable[[Module, int], tuple[str, Figure]]
-InputData = Union[Tensor, dict[str, Tensor]]
+InputData = Tensor | dict[str, Tensor]
 
 
 def print_metrics(loss: float | None, metrics: dict, iteration: int) -> None: