[hf_api] Fix incorrect typing

fadi212 · Dec 22, 2020 · 5b5f7dd · 5b5f7dd
1 parent 1558d19
commit 5b5f7dd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/transformers/hf_api.py b/src/transformers/hf_api.py
@@ -214,7 +214,7 @@ def model_list(self) -> List[ModelInfo]:
         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.