Skip to content

pipeline.py

get_dlc_root_data_dir()

Returns a list of root directories for Element DeepLabCut

Source code in workflow_deeplabcut/paths.py 
 5
 6
 7
 8
 9
10
11
12
13
def get_dlc_root_data_dir() -> list:
    """Returns a list of root directories for Element DeepLabCut"""
    dlc_root_dirs = dj.config.get("custom", {}).get("dlc_root_data_dir")
    if not dlc_root_dirs:
        return None
    elif not isinstance(dlc_root_dirs, abc.Sequence):
        return list(dlc_root_dirs)
    else:
        return dlc_root_dirs

get_dlc_processed_data_dir()

Returns an output directory relative to custom 'dlc_output_dir' root

Source code in workflow_deeplabcut/paths.py 
16
17
18
19
20
21
22
23
24
def get_dlc_processed_data_dir() -> str:
    """Returns an output directory relative to custom 'dlc_output_dir' root"""
    from pathlib import Path

    dlc_output_dir = dj.config.get("custom", {}).get("dlc_output_dir")
    if dlc_output_dir:
        return Path(dlc_output_dir)
    else:
        return None