Coverage for src/bob/bio/base/script/sort.py: 100%

1"""Sorts score files based on their score value 

2""" 

3import logging 

4 

5import click 

6import numpy 

7 

8from clapper.click import log_parameters, verbosity_option 

9 

10from bob.bio.base.score.load import dump_score, load_score 

11 

12logger = logging.getLogger(__name__) 

13 

14 

15@click.command( 

16 epilog="""\b 

17Examples: 

18 

19 $ bob bio sort -vvv /path/to/scores 

20""" 

21) 

22@click.argument( 

23 "score_paths", 

24 type=click.Path(exists=True, file_okay=True, dir_okay=False, writable=True), 

25 nargs=-1, 

26) 

27@verbosity_option(logger=logger) 

28def sort(score_paths, **kwargs): 

29 """Sorts score files based on their score values 

30 

31 The conversion happens in-place; backup your scores before using this script 

32 """ 

33 log_parameters(logger) 

34 

35 for path in score_paths: 

36 logger.info("Sorting: %s", path) 

37 scores = load_score(path) 

38 scores = scores[numpy.argsort(scores["score"])] 

39 dump_score(path, scores)