[hf_api] Fix incorrect typing

2020-12-22 19:52:47 +01:00
parent 1558d191e6
commit 5b5f7dd09c
1 changed files with 1 additions and 1 deletions
--- a/src/transformers/hf_api.py
+++ b/src/transformers/hf_api.py
@@ -214,7 +214,7 @@ class HfApi:
        d = r.json()
        return [ModelInfo(**x) for x in d]

-    def list_repos_objs(self, token: str, organization: Optional[str] = None) -> List[S3Obj]:
+    def list_repos_objs(self, token: str, organization: Optional[str] = None) -> List[RepoObj]:
        """
        HuggingFace git-based system, used for models.