Merge remote-tracking branch 'refs/remotes/origin/gt/ml-flow' into gt…

…/ml-flow
pasqal-io · Jul 17, 2024 · fdf05f1 · fdf05f1
2 parents 7df3c35 + aa98dfc
commit fdf05f1
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