Coverage for src/shephex/cli/slurm/add_info.py: 100%

15 statements  

« prev     ^ index     » next       coverage.py v7.6.1, created at 2025-06-20 14:13 +0200

1from pathlib import Path 

2 

3import rich_click as click 

4 

5from shephex.cli.slurm.slurm import slurm 

6from shephex.experiment import Experiment, ExperimentContext 

7 

8 

9@slurm.command() 

10@click.option("-j", "--job-id", type=str, required=True) 

11@click.option("-d", "--directory", type=click.Path(exists=True, file_okay=False, dir_okay=True), required=True) 

12def add_info(job_id: str, directory: click.Path) -> None: 

13 """ 

14 Add information to a job 

15 """ 

16 directory = Path(directory) 

17 context = ExperimentContext(directory / Experiment.shep_dir) 

18 try: 

19 context.meta.load(directory / Experiment.shep_dir) 

20 except FileNotFoundError: 

21 pass 

22 context.add('job-id', job_id)