Skip to content

Commit 3a06369

Browse files
authored
Add repr() to ModelFile and RepoObj (#75)
1 parent 1472e80 commit 3a06369

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

src/huggingface_hub/hf_api.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -49,6 +49,10 @@ def __init__(self, **kwargs):
4949
for k, v in kwargs.items():
5050
setattr(self, k, v)
5151

52+
def __repr__(self):
53+
items = (f"{k}='{v}'" for k, v in self.__dict__.items())
54+
return f"{self.__class__.__name__}({', '.join(items)})"
55+
5256

5357
class ModelFile:
5458
"""
@@ -60,6 +64,10 @@ def __init__(self, rfilename: str, **kwargs):
6064
for k, v in kwargs.items():
6165
setattr(self, k, v)
6266

67+
def __repr__(self):
68+
items = (f"{k}='{v}'" for k, v in self.__dict__.items())
69+
return f"{self.__class__.__name__}({', '.join(items)})"
70+
6371

6472
class ModelInfo:
6573
"""

0 commit comments

Comments
 (0)