make the docstring more readable and improve the list_models logic

Signed-off-by: devops117 <55235206+devops117@users.noreply.github.com>
This commit is contained in:
devops117 2022-11-23 10:40:27 +05:30 committed by Lincoln Stein
parent fdf41cc739
commit 248068fe5d

View File

@ -125,15 +125,18 @@ class ModelCache(object):
def list_models(self) -> dict:
'''
Return a dict of models in the format:
{ model_name1: {'status': ('active'|'cached'|'not loaded'),
{
model_name1: {
'status': ('active'|'cached'|'not loaded'),
'description': description,
},
model_name2: { etc }
model_name2: { etc },
}
'''
result = dict()
for name in self.config:
models = {}
for name, config in self.config.items():
try:
description = self.config[name].description
description = config.description
except ConfigAttributeError:
description = '<no description>'
@ -144,11 +147,13 @@ class ModelCache(object):
else:
status = 'not loaded'
result[name]={
models = models.update(
name = {
'status': status,
'description' : description
}
return result
'description': description,
})
return models
def print_models(self) -> None:
'''