Skip to content

Commit b0a42ad

Browse files
FixCardData.get() to respect default values when None (#2770)
* fix getter in CardData * typo * fix
1 parent 8aa1623 commit b0a42ad

File tree

2 files changed

+8
-1
lines changed

2 files changed

+8
-1
lines changed

src/huggingface_hub/repocard_data.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -221,7 +221,8 @@ def __str__(self):
221221

222222
def get(self, key: str, default: Any = None) -> Any:
223223
"""Get value for a given metadata key."""
224-
return self.__dict__.get(key, default)
224+
value = self.__dict__.get(key)
225+
return default if value is None else value
225226

226227
def pop(self, key: str, default: Any = None) -> Any:
227228
"""Pop value for a given metadata key."""

tests/test_repocard_data.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -63,6 +63,12 @@ def test_metadata_behave_as_dict(self):
6363
self.assertTrue("foo" in metadata)
6464
self.assertFalse("FOO" in metadata)
6565

66+
# default value
67+
# Should return default when key is not in metadata
68+
self.assertEqual(metadata.get("FOO", "default"), "default")
69+
# Should return default when key is in metadata but value is None
70+
metadata.FOO = None
71+
self.assertEqual(metadata.get("FOO", "default"), "default")
6672
# export
6773
self.assertEqual(str(metadata), "foo: BAR")
6874

0 commit comments

Comments
 (0)