mirror of
https://github.com/Doctorado-ML/benchmark.git
synced 2025-08-15 23:45:54 +00:00
Add -n to list
This commit is contained in:
@@ -701,7 +701,12 @@ class Summary:
|
||||
self.data.append(entry)
|
||||
|
||||
def list_results(
|
||||
self, score=None, model=None, input_data=None, sort_key="date"
|
||||
self,
|
||||
score=None,
|
||||
model=None,
|
||||
input_data=None,
|
||||
sort_key="date",
|
||||
number=0,
|
||||
) -> None:
|
||||
"""Print the list of results"""
|
||||
data = self.data.copy() if input_data is None else input_data
|
||||
@@ -716,6 +721,8 @@ class Summary:
|
||||
else x[sort_key],
|
||||
reverse=True,
|
||||
)
|
||||
if number > 0:
|
||||
data = data[:number]
|
||||
max_file = max(len(x["file"]) for x in data)
|
||||
max_title = max(len(x["title"]) for x in data)
|
||||
print(TextColor.LINE1, end="")
|
||||
|
14
src/list.py
14
src/list.py
@@ -37,13 +37,21 @@ def parse_arguments():
|
||||
default="date",
|
||||
help="key to sort results",
|
||||
)
|
||||
ap.add_argument(
|
||||
"-n",
|
||||
"--number",
|
||||
type=int,
|
||||
required=False,
|
||||
default=0,
|
||||
help="number of results to show, 0 to any",
|
||||
)
|
||||
args = ap.parse_args()
|
||||
|
||||
return (args.excel, args.score, args.model, args.key)
|
||||
return (args.excel, args.score, args.model, args.key, args.number)
|
||||
|
||||
|
||||
(excel, score, model, key) = parse_arguments()
|
||||
(excel, score, model, key, number) = parse_arguments()
|
||||
|
||||
data = Summary()
|
||||
data.acquire()
|
||||
data.list_results(score=score, model=model, sort_key=key)
|
||||
data.list_results(score=score, model=model, sort_key=key, number=number)
|
||||
|
Reference in New Issue
Block a user