mirror of
https://github.com/Doctorado-ML/benchmark.git
synced 2025-08-17 16:35:54 +00:00
Add -n to list
This commit is contained in:
@@ -701,7 +701,12 @@ class Summary:
|
|||||||
self.data.append(entry)
|
self.data.append(entry)
|
||||||
|
|
||||||
def list_results(
|
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:
|
) -> None:
|
||||||
"""Print the list of results"""
|
"""Print the list of results"""
|
||||||
data = self.data.copy() if input_data is None else input_data
|
data = self.data.copy() if input_data is None else input_data
|
||||||
@@ -716,6 +721,8 @@ class Summary:
|
|||||||
else x[sort_key],
|
else x[sort_key],
|
||||||
reverse=True,
|
reverse=True,
|
||||||
)
|
)
|
||||||
|
if number > 0:
|
||||||
|
data = data[:number]
|
||||||
max_file = max(len(x["file"]) for x in data)
|
max_file = max(len(x["file"]) for x in data)
|
||||||
max_title = max(len(x["title"]) for x in data)
|
max_title = max(len(x["title"]) for x in data)
|
||||||
print(TextColor.LINE1, end="")
|
print(TextColor.LINE1, end="")
|
||||||
|
14
src/list.py
14
src/list.py
@@ -37,13 +37,21 @@ def parse_arguments():
|
|||||||
default="date",
|
default="date",
|
||||||
help="key to sort results",
|
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()
|
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 = Summary()
|
||||||
data.acquire()
|
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