Commit fc24a93e authored by Julien Chaumond's avatar Julien Chaumond
Browse files

[HfApi] Add support for pipeline_tag

parent 0a3d0e02
...@@ -80,6 +80,7 @@ class ModelInfo: ...@@ -80,6 +80,7 @@ class ModelInfo:
author: Optional[str] = None, author: Optional[str] = None,
downloads: Optional[int] = None, downloads: Optional[int] = None,
tags: List[str] = [], tags: List[str] = [],
pipeline_tag: Optional[str] = None,
siblings: Optional[List[Dict]] = None, # list of files that constitute the model siblings: Optional[List[Dict]] = None, # list of files that constitute the model
**kwargs **kwargs
): ):
...@@ -88,6 +89,7 @@ class ModelInfo: ...@@ -88,6 +89,7 @@ class ModelInfo:
self.author = author self.author = author
self.downloads = downloads self.downloads = downloads
self.tags = tags self.tags = tags
self.pipeline_tag = pipeline_tag
self.siblings = [S3Object(**x) for x in siblings] if siblings is not None else None self.siblings = [S3Object(**x) for x in siblings] if siblings is not None else None
for k, v in kwargs.items(): for k, v in kwargs.items():
setattr(self, k, v) setattr(self, k, v)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment