From f5305c8e8f9030d9c11dc86f62519baf28637037 Mon Sep 17 00:00:00 2001 From: Nikita Date: Tue, 1 Oct 2024 11:22:02 +0000 Subject: [PATCH] fix: return type --- luxonis_train/core/utils/infer_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/luxonis_train/core/utils/infer_utils.py b/luxonis_train/core/utils/infer_utils.py index 630d2ccc..ded13b41 100644 --- a/luxonis_train/core/utils/infer_utils.py +++ b/luxonis_train/core/utils/infer_utils.py @@ -15,7 +15,7 @@ def render_visualizations( visualizations: dict[str, dict[str, Tensor]], save_dir: str | Path | None, show: bool = True, -) -> None: +) -> dict[str, list[np.ndarray]]: """Render or save visualizations.""" save_dir = Path(save_dir) if save_dir is not None else None if save_dir is not None: