@@ -28,7 +28,7 @@ class ModelInfo:
28
28
29
29
id : str
30
30
created : int
31
- owner : str
31
+ owned_by : str
32
32
object : str = "model"
33
33
34
34
@@ -56,9 +56,9 @@ def retrieve_model_info(args, model_id: str) -> Union[ModelInfo, None]:
56
56
if is_model_downloaded (model_id , args .model_directory ):
57
57
path = args .model_directory / model_config .name
58
58
created = int (os .path .getctime (path ))
59
- owner = getpwuid (os .stat (path ).st_uid ).pw_name
59
+ owned_by = getpwuid (os .stat (path ).st_uid ).pw_name
60
60
61
- return ModelInfo (id = model_config .name , created = created , owner = owner )
61
+ return ModelInfo (id = model_config .name , created = created , owned_by = owned_by )
62
62
return None
63
63
return None
64
64
@@ -79,8 +79,10 @@ def get_model_info_list(args) -> ModelInfo:
79
79
if is_model_downloaded (model_id , args .model_directory ):
80
80
path = args .model_directory / model_config .name
81
81
created = int (os .path .getctime (path ))
82
- owner = getpwuid (os .stat (path ).st_uid ).pw_name
82
+ owned_by = getpwuid (os .stat (path ).st_uid ).pw_name
83
83
84
- data .append (ModelInfo (id = model_config .name , created = created , owner = owner ))
84
+ data .append (
85
+ ModelInfo (id = model_config .name , created = created , owned_by = owned_by )
86
+ )
85
87
response = ModelInfoList (data = data )
86
88
return response
0 commit comments