From fee46267556502ab067c2c48e26ff7347b83f9d6 Mon Sep 17 00:00:00 2001
From: Giorgio Tosti Balducci
 <64412197+debrevitatevitae@users.noreply.github.com>
Date: Wed, 17 Jul 2024 22:24:59 +0200
Subject: [PATCH] Fix typing of InputData type

Co-authored-by: Roland-djee <9250798+Roland-djee@users.noreply.github.com>
---
 qadence/ml_tools/printing.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/qadence/ml_tools/printing.py b/qadence/ml_tools/printing.py
index f9fcabad..7d301d04 100644
--- a/qadence/ml_tools/printing.py
+++ b/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: