diff --git a/corrlib/cli.py b/corrlib/cli.py index b808c13..ff74527 100644 --- a/corrlib/cli.py +++ b/corrlib/cli.py @@ -90,6 +90,11 @@ def find( "--dataset", "-d", ), + arg: str = typer.Option( + str('all'), + "--show", + "-s", + ), ensemble: str = typer.Argument(), corr: str = typer.Argument(), code: str = typer.Argument(), @@ -98,8 +103,12 @@ def find( Find a record in the backlog at hand. Through specifying it's ensemble and the measured correlator. """ results = find_record(path, ensemble, corr, code) - print(results) - + if arg == "all": + print(results) + else: + for i in range(len(results)): + print(results[arg].values[i]) + return @app.command() def importer(