Merge pull request datajoint#3 from kushalbakshi/lu-lab

Use `output_dir` as Path object in `save_yaml`
kushalbakshi · Aug 14, 2024 · d76ecf5 · d76ecf5
2 parents 353315d + 9bd7e19
commit d76ecf5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/element_deeplabcut/readers/dlc_reader.py b/element_deeplabcut/readers/dlc_reader.py
@@ -238,7 +238,7 @@ def save_yaml(
     if "config_template" in config_dict:  # if passed full model.Model dict
         config_dict = config_dict["config_template"]
     if mkdir:
-        output_dir.mkdir(exist_ok=True)
+        Path(output_dir).mkdir(exist_ok=True)
     if "." in filename:  # if user provided extension, remove
         filename = filename.split(".")[0]