Fix typing in log_model_mlflow

Co-authored-by: Roland-djee <[email protected]>
pasqal-io · Jul 17, 2024 · aa98dfc · aa98dfc
1 parent fee4626
commit aa98dfc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/qadence/ml_tools/printing.py b/qadence/ml_tools/printing.py
@@ -82,7 +82,7 @@ def plot_mlflow(
 def log_model_mlflow(
     writer: Any,
     model: Module,
-    dataloader: Union[None, DataLoader, DictDataLoader],
+    dataloader: DataLoader | DictDataLoader | None
 ) -> None:
     if dataloader is not None:
         xs: InputData