Test for the presence of snapshot files when using modelzoo by jeylau · Pull Request #2485 · DeepLabCut/DeepLabCut

@jeylau

  • Test early whether superanimal name 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

@jeylau

@AlexEMG

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

AlexEMG

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Update comment as outlined.

@jeylau

MMathisLab

n-poulsen

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Looks good to me!