◐ Shell
clean mode source ↗

Add additional drop_likelihood_columns guards by C-Achard · Pull Request #3333 · DeepLabCut/DeepLabCut

Rename the private _drop_likelihood_columns to drop_likelihood_columns and use it across the codebase to consistently sanitize annotation DataFrames (drop any coord level named 'likelihood'). Updated imports/usages in modelzoo (ma_dlc, ma_dlc_dataframe), pose_estimation_pytorch (dlcloader), and utils (skeleton). Added a docstring note about centralizing project I/O and maintenance concerns. Added a unit test (tests/pose_estimation_pytorch/data/test_dlc_dataloader.py) to ensure DLCLoader.to_coco ignores likelihood columns.

@C-Achard self-assigned this

May 18, 2026

@C-Achard C-Achard linked an issue

May 18, 2026

that may be closed by this pull request

Change logging from info to warning (with stacklevel=2) when likelihood columns are detected and drop them. Call drop_likelihood_columns immediately after reading the HDF file. Normalize variable naming by replacing Data with data throughout mergeandsplit and update subsequent references (index ranges, scorer extraction, and enumeration) to use the lower-case variable.

deruyter92

@C-Achard

@C-Achard C-Achard marked this pull request as ready for review

May 19, 2026 09:33

@C-Achard

@C-Achard C-Achard deleted the cy/fix-likelihood-coco branch

May 20, 2026 12:16