fix: shi

ape_addressbook/addressbook.py

@@ -26,12 +26,16 @@ def _validate_entries(entries: Dict) -> Dict:
 
 
 class AddressBookConfig(PluginConfig):
-    __root__: Dict[str, AddressType] = {}
-
     @root_validator(pre=True)
     def validate_entries(cls, entries):
         return _validate_entries(entries)
 
+    def __len__(self) -> int:
+        return len(self.dict())
+
+    class Config:
+        extra = "allow"
+
 
 class AddressBook(ManagerAccessMixin):
     """
@@ -67,14 +71,18 @@ def registry(self) -> Dict[str, AddressType]:
         and project addresses.
         """
 
-        return self.config.dict()
+        data = self.config.dict()
+
+        # Sorted for consistency's sake.
+        return {k: data[k] for k in sorted(data)}
 
     @property
     def aliases(self) -> Iterator[str]:
         """
         An iterator over all aliases in the registry.
         """
 
+        # NOTE: self.registry is sorted.
         for alias in self.registry:
             yield alias
 

setup.py

@@ -58,7 +58,7 @@
     include_package_data=True,
     install_requires=[
         "eth-ape>=0.6.0,<0.7",
-        "pydantic<2",
+        "pydantic",  # Use same version as eth-ape.
     ],
     python_requires=">=3.8,<4",
     extras_require=extras_require,

tests/test_addressbook.py

@@ -29,10 +29,9 @@ def test_aliases(project_alias_unchecksummed, project_alias_no_quotes):
     The aliases includes both project and global addresses.
     """
 
-    assert list(addressbook.aliases) == [
-        project_alias_unchecksummed,
-        project_alias_no_quotes,
-    ]
+    actual = list(addressbook.aliases)
+    expected = [project_alias_no_quotes, project_alias_unchecksummed]
+    assert actual == expected
 
 
 def test_contains(