Merge pull request #75 from ttngu207/main

fix bug
datajoint · Mar 7, 2023 · 40503d7 · 40503d7
2 parents 1627433 + 37a100c
commit 40503d7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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: