Merge pull request #129 from ttngu207/main

datajoint · Sep 20, 2024 · f7cac84 · f7cac84
2 parents af4ce80 + 144c34a
commit f7cac84
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/element_deeplabcut/plotting/dlc_plots.py b/element_deeplabcut/plotting/dlc_plots.py
@@ -9,14 +9,15 @@ def plotting_results(pose_estimation_key: dict):
         plots_dir (Path): Path to the folder containing the plots
     """
     import deeplabcut
-    from element_deeplabcut import dlc_reader, model
+    from element_deeplabcut import model
+    from element_deeplabcut.readers import dlc_reader
 
     output_dir = (model.PoseEstimationTask & pose_estimation_key).fetch1(
         "pose_estimation_output_dir"
     )
     output_dir = model.find_full_path(model.get_dlc_root_data_dir(), output_dir)
 
-    dlc_result = dlc_reader.PoseEstimation(output_dir)
+    dlc_result = dlc_reader.PoseEstimation(output_dir.as_posix())
 
     plots_dir = output_dir / "plots"
     plots_dir.mkdir(exist_ok=True)