tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

fix DefaultDict.__or__

Browse Source
This commit is contained in:
Aman Karmani
2023-08-10 03:56:50 +00:00
parent 918f1b0dfb
commit a13e45d548
2 changed files with 10 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/axolotl/utils/dict.py
View File

@@ -10,3 +10,6 @@ class DictDefault(Dict):
def __missing__(self, key): def __missing__(self, key):
return None return None
def __or__(self, other):
return DictDefault(super().__or__(other))

7
tests/test_dict.py
View File

@@ -72,6 +72,13 @@ class DictDefaultTest(unittest.TestCase):
assert cfg.random_key is None, "DictDefault should return None for missing keys" assert cfg.random_key is None, "DictDefault should return None for missing keys"
def test_dict_or(self):
cfg = DictDefault({}) | DictDefault({})
assert (
cfg.random_key is None
), "DictDefault should return None for missing keys after | operation"
def test_dict_nested_missingparentkey(self): def test_dict_nested_missingparentkey(self):
""" """
Due to subclassing Dict, DictDefault will error if we try to access a nested key whose parent key does not exist. Due to subclassing Dict, DictDefault will error if we try to access a nested key whose parent key does not exist.