Test for the presence of snapshot files when using modelzoo by jeylau · Pull Request #2485 · DeepLabCut/DeepLabCut
- Test early whether
superanimalname is a valid option - Test for the presence of model weight files rather than the only existence of a weight folder (so if the folder exists, but it does not contain any model snapshots, superanimal model weights will–as they should–be redownloaded, preventing the IndexError reported in other issues.)
Fixes #2469
Add comment here: raise ValueError(f"{superanimal_name} not available. Available ones are: ", MODELOPTIONS)
IF you see the model you want, go to the website and see what is available/upldate dlclib
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Update comment as outlined.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Looks good to me!
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters. Learn more about bidirectional Unicode characters