From 37a100c0c07042cb4fecda355336687d445f8db7 Mon Sep 17 00:00:00 2001 From: Thinh Nguyen Date: Thu, 2 Mar 2023 19:21:28 -0600 Subject: [PATCH] fix bug --- element_deeplabcut/model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/element_deeplabcut/model.py b/element_deeplabcut/model.py index f170ad8..ef49333 100644 --- a/element_deeplabcut/model.py +++ b/element_deeplabcut/model.py @@ -630,7 +630,7 @@ def generate( """ processed_dir = get_dlc_processed_data_dir() output_dir = cls.infer_output_dir( - video_recording_key, relative=False, mkdir=True + {**video_recording_key, "model_name": model_name}, relative=False, mkdir=True ) if task_mode is None: